diff --git a/ax/service/utils/report_utils.py b/ax/service/utils/report_utils.py index e4ea9b935e1..fd563ac0f48 100644 --- a/ax/service/utils/report_utils.py +++ b/ax/service/utils/report_utils.py @@ -391,7 +391,10 @@ def get_standard_plots( if global_sensitivity_analysis and isinstance(model, TorchModelBridge): try: sens = ax_parameter_sens(model, order="total") - importance_measure = 'Variance-based sensitivity analysis' + importance_measure = ( + 'Variance-based sensitivity analysis' + ) except NotImplementedError as e: logger.info(f"Failed to compute global feature sensitivities: {e}") feature_importance_plot = plot_feature_importance_by_feature_plotly( diff --git a/sphinx/source/models.rst b/sphinx/source/models.rst index f520856d90c..620284be08f 100644 --- a/sphinx/source/models.rst +++ b/sphinx/source/models.rst @@ -264,6 +264,14 @@ ax.models.torch.botorch_modular.utils module :undoc-members: :show-inheritance: +ax.models.torch.botorch_modular.kernels module +~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ + +.. automodule:: ax.models.torch.botorch_modular.kernels + :members: + :undoc-members: + :show-inheritance: + ax.models.torch.cbo_lcea module ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~