Fix generation node tests

Summary: A couple tests were broken in a clean up diff yesterday. Somehow slipped through the CI. Differential Revision: D55753520
facebook · Apr 4, 2024 · 8782339 · 8782339
1 parent c7601b4
commit 8782339
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]