Skip to content

Commit 1b12c7d

Browse files
committed
Fix style
1 parent 213644f commit 1b12c7d

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

optimas/generators/ax/developer/multitask.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -85,14 +85,13 @@ def get_MTGP(
8585
device: torch.device = torch.device("cpu"),
8686
dtype: torch.dtype = torch.double,
8787
) -> TorchModelBridge:
88-
"""Instantiates a Multi-task Gaussian Process (MTGP) model that generates
89-
points with EI.
88+
"""Instantiate a Multi-task Gaussian Process (MTGP) model.
9089
90+
Points are generated with EI (Expected Improvement).
9191
If the input experiment is a MultiTypeExperiment then a
9292
Multi-type Multi-task GP model will be instantiated.
9393
Otherwise, the model will be a Single-type Multi-task GP.
9494
"""
95-
9695
if isinstance(experiment, MultiTypeExperiment):
9796
trial_index_to_type = {
9897
t.index: t.trial_type for t in experiment.trials.values()

0 commit comments

Comments
 (0)