From ee9fa033b2dd00a21a42734b36eefca37613b476 Mon Sep 17 00:00:00 2001
From: Amani Chabouni <142830128+Amanichab@users.noreply.github.com>
Date: Thu, 26 Jun 2025 21:29:15 -0400
Subject: [PATCH 1/2] Update benchmark_assignment.py

Changed Models to Generators
---
 benchmark_assignment.py | 12 ++++++------
 1 file changed, 6 insertions(+), 6 deletions(-)

diff --git a/benchmark_assignment.py b/benchmark_assignment.py
index 836b546..7c36053 100644
--- a/benchmark_assignment.py
+++ b/benchmark_assignment.py
@@ -18,8 +18,8 @@
 
 import numpy as np
 from ax.service.ax_client import AxClient, ObjectiveProperties
-from ax.modelbridge.factory import Models
-from ax.modelbridge.generation_strategy import GenerationStep, GenerationStrategy
+from ax.modelbridge.factory import Generators
+from ax.generation_strategy.generation_strategy import GenerationStep, GenerationStrategy
 from botorch.acquisition import ExpectedImprovement
 
 seed_list = [0, 1, 2, 3, 4, 5, 6, 7, 8, 9]
@@ -42,8 +42,8 @@
 # --------------------------------------------------------------------------------------
 import numpy as np
 from ax.service.ax_client import AxClient, ObjectiveProperties
-from ax.modelbridge.factory import Models
-from ax.modelbridge.generation_strategy import GenerationStep, GenerationStrategy
+from ax.modelbridge.factory import Generators
+from ax.generation_strategy.generation_strategy import GenerationStep, GenerationStrategy
 from botorch.acquisition import UpperConfidenceBound
 
 seed_list = [0, 1, 2, 3, 4, 5, 6, 7, 8, 9]
@@ -81,8 +81,8 @@
 
 import numpy as np
 from ax.service.ax_client import AxClient, ObjectiveProperties
-from ax.modelbridge.factory import Models
-from ax.modelbridge.generation_strategy import GenerationStep, GenerationStrategy
+from ax.modelbridge.factory import Generators
+from ax.generation_strategy.generation_strategy import GenerationStep, GenerationStrategy
 from botorch.acquisition import LogExpectedImprovement
 
 seed_list = [0, 1, 2, 3, 4, 5, 6, 7, 8, 9]

From 3be1f5424df382799a0c48369408ea3fd0409dbe Mon Sep 17 00:00:00 2001
From: Amani Chabouni <142830128+Amanichab@users.noreply.github.com>
Date: Thu, 26 Jun 2025 21:30:10 -0400
Subject: [PATCH 2/2] Update README.md

Changed Models to Generators
---
 README.md | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/README.md b/README.md
index 7a39791..c81ed90 100644
--- a/README.md
+++ b/README.md
@@ -43,7 +43,7 @@ An example of how to specify such models is provided in the snippet below:
 ```python
 ...
     GenerationStep(
-        model=Models.BOTORCH_MODULAR,
+        model=Generators.BOTORCH_MODULAR,
         num_trials=-1,
         max_parallelism=3,
         model_kwargs={"botorch_acqf_class": qKnowledgeGradient},