From f371f2c9d417ecfc10d6a91e5df1719fab374ffd Mon Sep 17 00:00:00 2001
From: Amani Chabouni <142830128+Amanichab@users.noreply.github.com>
Date: Thu, 26 Jun 2025 21:14:16 -0400
Subject: [PATCH 1/2] Update requirements.txt

---
 requirements.txt | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/requirements.txt b/requirements.txt
index 5e8874d..9757473 100644
--- a/requirements.txt
+++ b/requirements.txt
@@ -1,4 +1,4 @@
 -e . # package from this repo
-ax-platform
+ax-platform<1
 matplotlib
 pytest

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

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

diff --git a/MTBO_assignment.py b/MTBO_assignment.py
index 7d316f0..3d54ab6 100644
--- a/MTBO_assignment.py
+++ b/MTBO_assignment.py
@@ -19,8 +19,8 @@
 
 import numpy as np
 from ax.core.observation import ObservationFeatures
-from ax.modelbridge.generation_strategy import GenerationStep, GenerationStrategy
-from ax.modelbridge.registry import Models
+from ax.modelbridge.factory import Generators
+from ax.generation_strategy.generation_strategy import GenerationStep, GenerationStrategy
 from ax.modelbridge.transforms.task_encode import TaskEncode
 from ax.modelbridge.transforms.unit_x import UnitX
 from ax.service.ax_client import AxClient, ObjectiveProperties
@@ -31,12 +31,12 @@
     name="MultiTaskOp",
     steps=[
         GenerationStep(
-            model=Models.SOBOL,
+            model=Generators.SOBOL,
             num_trials=10,
             model_kwargs={"deduplicate": True, "transforms": transforms},
         ),
         GenerationStep(
-            model=Models.BOTORCH_MODULAR,
+            model=Generators.BOTORCH_MODULAR,
             num_trials=-1,
             model_kwargs={"transforms": transforms},
         ),
@@ -83,11 +83,11 @@
     name="MultiTaskOp",
     steps=[
         GenerationStep(
-            model=Models.SOBOL,
+            model=Generators.SOBOL,
             num_trials=10,
         ),
         GenerationStep(
-            model=Models.BOTORCH_MODULAR,
+            model=Generators.BOTORCH_MODULAR,
             num_trials=-1,
         ),
     ],