@@ -196,17 +196,12 @@ def __init__(
196
196
model_save_period : Optional [int ] = 5 ,
197
197
model_history_dir : Optional [str ] = "model_history" ,
198
198
) -> None :
199
- # SH note for standardization
200
199
# As trial parameters these get written to history array
201
200
# Ax trial_index and arm toegther locate a point
202
201
# Multiple points (Optimas trials) can share the same Ax trial_index
203
- # Standard (VOCS) does not have equiv. of trial parameters -> would be variables.
204
- # If want to use _id can have a mapping inside the generator, but those
205
- # points that are part of same ax trial_index will not be seen in history.
206
202
custom_trial_parameters = [
207
203
TrialParameter ("arm_name" , "ax_arm_name" , dtype = "U32" ),
208
204
TrialParameter ("trial_type" , "ax_trial_type" , dtype = "U32" ),
209
- # SH changed from trial_index to ax_trial_id - trial_index is usually Optimas index?
210
205
TrialParameter ("ax_trial_id" , "ax_trial_index" , dtype = int ),
211
206
]
212
207
self ._check_inputs (varying_parameters , objectives , lofi_task , hifi_task )
@@ -289,7 +284,7 @@ def suggest(self, num_points: Optional[int]) -> List[dict]:
289
284
290
285
def ingest (self , results : List [dict ]) -> None :
291
286
"""Incorporate evaluated trials into experiment."""
292
- # reconstruct Optimastrials
287
+ # reconstruct Optimas trials
293
288
trials = []
294
289
for result in results :
295
290
trial = Trial .from_dict (
0 commit comments