File tree Expand file tree Collapse file tree 1 file changed +2
-6
lines changed
optimas/generators/ax/developer Expand file tree Collapse file tree 1 file changed +2
-6
lines changed Original file line number Diff line number Diff line change @@ -202,9 +202,7 @@ def __init__(
202
202
TrialParameter ("trial_type" , "ax_trial_type" , dtype = "U32" ),
203
203
TrialParameter ("trial_index" , "ax_trial_index" , dtype = int ),
204
204
]
205
- self ._check_inputs (
206
- varying_parameters , objectives , lofi_task , hifi_task
207
- )
205
+ self ._check_inputs (varying_parameters , objectives , lofi_task , hifi_task )
208
206
super ().__init__ (
209
207
varying_parameters = varying_parameters ,
210
208
objectives = objectives ,
@@ -349,9 +347,7 @@ def _complete_evaluations(self, trials: List[Trial]) -> None:
349
347
objective_eval = {}
350
348
oe = trial .objective_evaluations [0 ]
351
349
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
355
351
else :
356
352
self .current_trial .mark_arm_abandoned (trial .arm_name )
357
353
if trial .trial_type == self .lofi_task .name :
You can’t perform that action at this time.
0 commit comments