File tree Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Expand file tree Collapse file tree 1 file changed +2
-2
lines changed Original file line number Diff line number Diff line change @@ -818,7 +818,7 @@ def test_ax_service_init():
818
818
# are replaced by Manual trials.
819
819
df = ax_client .get_trials_data_frame ()
820
820
for j in range (i ):
821
- assert df ["generation_method" ][j ] == "Manual"
821
+ assert df ["generation_method" ][j ] is None
822
822
for k in range (i , n_init - 1 ):
823
823
assert df ["generation_method" ][k ] == "Sobol"
824
824
@@ -868,7 +868,7 @@ def test_ax_service_init():
868
868
# `n_external` Manual trials.
869
869
df = ax_client .get_trials_data_frame ()
870
870
for j in range (n_external ):
871
- assert df ["generation_method" ][j ] == "Manual"
871
+ assert df ["generation_method" ][j ] is None
872
872
for k in range (n_external , n_external + n_init ):
873
873
assert df ["generation_method" ][k ] == "Sobol"
874
874
df ["generation_method" ][n_external + n_init ] == "BoTorch"
You can’t perform that action at this time.
0 commit comments