From 3e388fa94f737c5ce09ff3406c084255c926f441 Mon Sep 17 00:00:00 2001
From: Jerry Lin <zylin@meta.com>
Date: Wed, 5 Feb 2025 16:22:46 -0800
Subject: [PATCH] Migrate GS usages to new module location (#623)

Summary:
Pull Request resolved: https://github.com/facebookresearch/aepsych/pull/623

X-link: https://github.com/facebook/Ax/pull/3311

Removing the following old files from `ax/modelbridge/` in favor of the new `ax/generation_strategy` directory.

```
best_model_selector.py
dispatch_utils.py
external_generation_node.py
generation_node_input_constructors.py
generation_node.py
generation_strategy.py
model_spec.py
transition_criterion.py
```

Reviewed By: saitcakmak

Differential Revision: D68645075
---
 website/pages/api/_modules/aepsych/generators/base.html | 2 +-
 website/pages/api/_modules/aepsych/strategy.html        | 2 +-
 2 files changed, 2 insertions(+), 2 deletions(-)

diff --git a/website/pages/api/_modules/aepsych/generators/base.html b/website/pages/api/_modules/aepsych/generators/base.html
index 841465177..63e5840dd 100644
--- a/website/pages/api/_modules/aepsych/generators/base.html
+++ b/website/pages/api/_modules/aepsych/generators/base.html
@@ -23,7 +23,7 @@ <h1>Source code for aepsych.generators.base</h1><div class="highlight"><pre>
 <span class="kn">from</span> <span class="nn">aepsych.config</span> <span class="kn">import</span> <span class="n">Config</span><span class="p">,</span> <span class="n">ConfigurableMixin</span>
 <span class="kn">from</span> <span class="nn">aepsych.models.base</span> <span class="kn">import</span> <span class="n">AEPsychMixin</span>
 <span class="kn">from</span> <span class="nn">ax.core.experiment</span> <span class="kn">import</span> <span class="n">Experiment</span>
-<span class="kn">from</span> <span class="nn">ax.modelbridge.generation_node</span> <span class="kn">import</span> <span class="n">GenerationStep</span>
+<span class="kn">from</span> <span class="nn">ax.generation_strategy.generation_node</span> <span class="kn">import</span> <span class="n">GenerationStep</span>
 <span class="kn">from</span> <span class="nn">botorch.acquisition</span> <span class="kn">import</span> <span class="p">(</span>
     <span class="n">AcquisitionFunction</span><span class="p">,</span>
     <span class="n">NoisyExpectedImprovement</span><span class="p">,</span>
diff --git a/website/pages/api/_modules/aepsych/strategy.html b/website/pages/api/_modules/aepsych/strategy.html
index 5595d851d..5a5fe0628 100644
--- a/website/pages/api/_modules/aepsych/strategy.html
+++ b/website/pages/api/_modules/aepsych/strategy.html
@@ -43,7 +43,7 @@ <h1>Source code for aepsych.strategy</h1>
 <span class="p">)</span>
 <span class="kn">from</span> <span class="nn">aepsych.utils_logging</span> <span class="kn">import</span> <span class="n">getLogger</span>
 <span class="kn">from</span> <span class="nn">ax.core.base_trial</span> <span class="kn">import</span> <span class="n">TrialStatus</span>
-<span class="kn">from</span> <span class="nn">ax.modelbridge.generation_strategy</span> <span class="kn">import</span> <span class="n">GenerationStrategy</span>
+<span class="kn">from</span> <span class="nn">ax.generation_strategy.generation_strategy</span> <span class="kn">import</span> <span class="n">GenerationStrategy</span>
 <span class="kn">from</span> <span class="nn">ax.plot.contour</span> <span class="kn">import</span> <span class="n">interact_contour</span>
 <span class="kn">from</span> <span class="nn">ax.plot.slice</span> <span class="kn">import</span> <span class="n">plot_slice</span>
 <span class="kn">from</span> <span class="nn">ax.service.ax_client</span> <span class="kn">import</span> <span class="n">AxClient</span>