Merge 8782339 into c7601b4

facebook · Apr 4, 2024 · 49c57a9 · 49c57a9
2 parents c7601b4 + 8782339
commit 49c57a9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/ax/modelbridge/tests/test_generation_node.py b/ax/modelbridge/tests/test_generation_node.py
@@ -313,7 +313,7 @@ def setUp(self) -> None:
  ms_gpei = ModelSpec(model_enum=Models.GPEI)
  ms_gpei.fit(experiment=self.branin_experiment, data=data)
 
- ms_botorch = ModelSpec(model_enum=Models.BOTORCH)
+ ms_botorch = ModelSpec(model_enum=Models.BOTORCH_MODULAR)
  ms_botorch.fit(experiment=self.branin_experiment, data=data)
 
  self.fitted_model_specs = [ms_gpei, ms_botorch]