Skip to content

Commit 012a69e

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent 0ba440f commit 012a69e

File tree

1 file changed

+2
-6
lines changed

1 file changed

+2
-6
lines changed

optimas/generators/ax/developer/multitask.py

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -202,9 +202,7 @@ def __init__(
202202
TrialParameter("trial_type", "ax_trial_type", dtype="U32"),
203203
TrialParameter("trial_index", "ax_trial_index", dtype=int),
204204
]
205-
self._check_inputs(
206-
varying_parameters, objectives, lofi_task, hifi_task
207-
)
205+
self._check_inputs(varying_parameters, objectives, lofi_task, hifi_task)
208206
super().__init__(
209207
varying_parameters=varying_parameters,
210208
objectives=objectives,
@@ -349,9 +347,7 @@ def _complete_evaluations(self, trials: List[Trial]) -> None:
349347
objective_eval = {}
350348
oe = trial.objective_evaluations[0]
351349
objective_eval["f"] = (oe.value, oe.sem)
352-
self.current_trial.run_metadata[trial.arm_name] = (
353-
objective_eval
354-
)
350+
self.current_trial.run_metadata[trial.arm_name] = objective_eval
355351
else:
356352
self.current_trial.mark_arm_abandoned(trial.arm_name)
357353
if trial.trial_type == self.lofi_task.name:

0 commit comments

Comments
 (0)