From 66660e341b7dd0780feac4640f3709a8fd024206 Mon Sep 17 00:00:00 2001 From: Sait Cakmak Date: Wed, 30 Oct 2024 07:00:46 -0700 Subject: [PATCH] Import fix in custom_botorch_model_in_ax tutorial (#2601) Summary: This optimizer mock was renamed in Ax, leading to an import failure in this tutorial. Pull Request resolved: https://github.com/pytorch/botorch/pull/2601 Reviewed By: Balandat, esantorella Differential Revision: D65205716 Pulled By: saitcakmak fbshipit-source-id: 488092748f4a4f6cf4173c05e3c2d6a4269ae95e --- tutorials/custom_botorch_model_in_ax.ipynb | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tutorials/custom_botorch_model_in_ax.ipynb b/tutorials/custom_botorch_model_in_ax.ipynb index d85edde763..4f687d77a1 100644 --- a/tutorials/custom_botorch_model_in_ax.ipynb +++ b/tutorials/custom_botorch_model_in_ax.ipynb @@ -34,7 +34,7 @@ "import os\n", "from contextlib import contextmanager, nullcontext\n", "\n", - "from ax.utils.testing.mock import fast_botorch_optimize_context_manager\n", + "from ax.utils.testing.mock import mock_botorch_optimize_context_manager\n", "import plotly.io as pio\n", "\n", "# Ax uses Plotly to produce interactive plots. These are great for viewing and analysis,\n", @@ -362,7 +362,7 @@ ], "source": [ "if SMOKE_TEST:\n", - " fast_smoke_test = fast_botorch_optimize_context_manager\n", + " fast_smoke_test = mock_botorch_optimize_context_manager\n", "else:\n", " fast_smoke_test = nullcontext\n", "\n",