Fix generation node tests (#2322)

Summary:
Pull Request resolved: #2322

A couple tests were broken in a clean up diff yesterday. Somehow slipped through the CI.

Reviewed By: dme65

Differential Revision: D55753520

fbshipit-source-id: 09d91600a0e2ec2394bfb25e61edb5e91e83b2d3

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]