Skip to content

Commit 5bf6d59

Browse files
committed
Update ax_model_manager to BOTORCH_MODULAR
1 parent 863fdae commit 5bf6d59

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

optimas/utils/ax/ax_model_manager.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -130,7 +130,7 @@ def _build_ax_client_from_dataframe(
130130
# allow calling `model.predict`. Using MOO for multiobjective is
131131
# needed because otherwise calls to `get_pareto_optimal_parameters`
132132
# would fail.
133-
model = Models.GPEI if len(objectives) == 1 else Models.MOO
133+
model = Models.BOTORCH_MODULAR
134134
gs = GenerationStrategy([GenerationStep(model=model, num_trials=-1)])
135135
ax_client = AxClient(generation_strategy=gs, verbose_logging=False)
136136
ax_client.create_experiment(

0 commit comments

Comments
 (0)