Change gen strategy from GPEI to BoTorch

Author: shuds13

tests/test_ax_generators.py

@@ -488,7 +488,7 @@ def test_ax_single_fidelity_resume():
             # Check that the sobol step has been skipped.
             df = ax_client.get_trials_data_frame()
             assert len(df) == 12
-            assert df["generation_method"].to_numpy()[-1] == "GPEI"
+            assert df["generation_method"].to_numpy()[-1] == "BoTorch"
 
             check_run_ax_service(
                 ax_client, gen, exploration, n_failed_expected=2
@@ -821,7 +821,8 @@ def test_ax_service_init():
             assert df["generation_method"][j] == "Manual"
         for k in range(i, n_init - 1):
             assert df["generation_method"][k] == "Sobol"
-        df["generation_method"][min(i, n_init)] == "GPEI"
+
+        df["generation_method"][min(i, n_init)] == "BoTorch"
 
     # Try to load saved client from json. This used to fail when the SOBOL
     # step was skipped due to n_external > n_init. It is added here to prevent
@@ -870,21 +871,30 @@ def test_ax_service_init():
         assert df["generation_method"][j] == "Manual"
     for k in range(n_external, n_external + n_init):
         assert df["generation_method"][k] == "Sobol"
-    df["generation_method"][n_external + n_init] == "GPEI"
+    df["generation_method"][n_external + n_init] == "BoTorch"
 
 
 if __name__ == "__main__":
-    test_ax_single_fidelity()
-    test_ax_single_fidelity_resume()
-    test_ax_single_fidelity_int()
-    test_ax_single_fidelity_moo()
-    test_ax_single_fidelity_fb()
-    test_ax_single_fidelity_moo_fb()
-    test_ax_single_fidelity_updated_params()
-    test_ax_multi_fidelity()
-    test_ax_multitask()
-    test_ax_client()
-    test_ax_single_fidelity_with_history()
-    test_ax_multi_fidelity_with_history()
-    test_ax_multitask_with_history()
-    test_ax_service_init()
+    tests = [
+        test_ax_single_fidelity,
+        test_ax_single_fidelity_resume,
+        test_ax_single_fidelity_int,
+        test_ax_single_fidelity_moo,
+        test_ax_single_fidelity_fb,
+        test_ax_single_fidelity_moo_fb,
+        test_ax_single_fidelity_updated_params,
+        test_ax_multi_fidelity,
+        test_ax_multitask,
+        test_ax_client,
+        test_ax_single_fidelity_with_history,
+        test_ax_multi_fidelity_with_history,
+        test_ax_multitask_with_history,
+        test_ax_service_init,
+    ]
+
+    for test in tests:
+        print()
+        print("-" * len(test.__name__))
+        print(test.__name__)  # Print test name
+        print("-" * len(test.__name__))
+        test()  # Run the test