Skip to content

Commit 07eee24

Browse files
[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
1 parent ce09d6b commit 07eee24

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

optimas/generators/ax/developer/multitask.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -223,7 +223,7 @@ def __init__(
223223
self.current_trial = None
224224
self.gr_lofi = None
225225
self._experiment = self._create_experiment()
226-
226+
227227
# Internal mapping: _id -> (arm_name, ax_trial_id, trial_type)
228228
self._id_mapping = {}
229229
self._next_id = 0
@@ -275,7 +275,7 @@ def suggest(self, num_points: Optional[int]) -> List[dict]:
275275
self._id_mapping[current_id] = {
276276
"arm_name": arm.name,
277277
"ax_trial_id": trial_index,
278-
"trial_type": trial_type
278+
"trial_type": trial_type,
279279
}
280280
point["_id"] = current_id
281281
point["trial_type"] = trial_type # Keep trial_type for now
@@ -296,15 +296,15 @@ def ingest(self, results: List[dict]) -> None:
296296
custom_parameters=self._custom_trial_parameters,
297297
)
298298
trials.append(trial)
299-
299+
300300
# Apply _id mapping to all trials before processing
301301
for trial in trials:
302302
if trial.gen_id is not None and trial.gen_id in self._id_mapping:
303303
mapping = self._id_mapping[trial.gen_id]
304304
trial.arm_name = mapping["arm_name"]
305305
trial.ax_trial_id = mapping["ax_trial_id"]
306306
# trial_type should already be in trial from custom_parameters
307-
307+
308308
if self.gen_state == NOT_STARTED:
309309
self._incorporate_external_data(trials)
310310
else:

0 commit comments

Comments
 (0)