From 8782339059786a9f1d8aea0ae39d6a678eef1811 Mon Sep 17 00:00:00 2001 From: Sait Cakmak Date: Thu, 4 Apr 2024 11:09:43 -0700 Subject: [PATCH] Fix generation node tests Summary: A couple tests were broken in a clean up diff yesterday. Somehow slipped through the CI. Differential Revision: D55753520 --- ax/modelbridge/tests/test_generation_node.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ax/modelbridge/tests/test_generation_node.py b/ax/modelbridge/tests/test_generation_node.py index 67b96316ccc..17d4e32d67f 100644 --- 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]