Skip to content

Commit fccfa18

Browse files
committed
Replace Manual with None
1 parent 4be14a7 commit fccfa18

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

tests/test_ax_generators.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -818,7 +818,7 @@ def test_ax_service_init():
818818
# are replaced by Manual trials.
819819
df = ax_client.get_trials_data_frame()
820820
for j in range(i):
821-
assert df["generation_method"][j] == "Manual"
821+
assert df["generation_method"][j] is None
822822
for k in range(i, n_init - 1):
823823
assert df["generation_method"][k] == "Sobol"
824824

@@ -868,7 +868,7 @@ def test_ax_service_init():
868868
# `n_external` Manual trials.
869869
df = ax_client.get_trials_data_frame()
870870
for j in range(n_external):
871-
assert df["generation_method"][j] == "Manual"
871+
assert df["generation_method"][j] is None
872872
for k in range(n_external, n_external + n_init):
873873
assert df["generation_method"][k] == "Sobol"
874874
df["generation_method"][n_external + n_init] == "BoTorch"

0 commit comments

Comments
 (0)