Skip to content

Commit c66e09f

Browse files
committed
Fix to use ax_trial_id from history points
1 parent 48c83e8 commit c66e09f

File tree

1 file changed

+3
-3
lines changed

1 file changed

+3
-3
lines changed

optimas/generators/ax/developer/multitask.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -308,18 +308,18 @@ def ingest(self, results: List[dict]) -> None:
308308
def _incorporate_external_data(self, trials: List[Trial]) -> None:
309309
"""Incorporate external data (e.g., from history) into experiment."""
310310
# Get trial indices.
311-
# SH should have handling if trial_indexs are None...
311+
# SH should have handling if ax_trial_ids are None...
312312
trial_indices = []
313313
for trial in trials:
314-
trial_indices.append(trial.trial_index)
314+
trial_indices.append(trial.ax_trial_id)
315315
trial_indices = np.unique(np.array(trial_indices))
316316

317317
# Group trials by index.
318318
grouped_trials = {}
319319
for index in trial_indices:
320320
grouped_trials[index] = []
321321
for trial in trials:
322-
grouped_trials[trial.trial_index].append(trial)
322+
grouped_trials[trial.ax_trial_id].append(trial)
323323

324324
# Add trials to experiment.
325325
for index in trial_indices:

0 commit comments

Comments
 (0)