Skip to content

Commit e858bae

Browse files
committed
Prevent duplicate point return
1 parent cd694e0 commit e858bae

File tree

1 file changed

+10
-10
lines changed

1 file changed

+10
-10
lines changed

optimas/generators/ax/developer/multitask.py

Lines changed: 10 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -275,16 +275,16 @@ def suggest(self, num_points: Optional[int]) -> List[dict]:
275275
next_trial = self._get_next_trial_arm()
276276
if next_trial is not None:
277277
arm, trial_type, trial_index = next_trial
278-
point = {
279-
var.name: arm.parameters.get(var.name)
280-
for var in self._varying_parameters
281-
}
282-
# SH for VOCS standard these will need to be declared as variables
283-
# For now much match the trial parameter names.
284-
point["ax_trial_id"] = trial_index
285-
point["arm_name"] = arm.name
286-
point["trial_type"] = trial_type
287-
points.append(point)
278+
point = {
279+
var.name: arm.parameters.get(var.name)
280+
for var in self._varying_parameters
281+
}
282+
# SH for VOCS standard these will need to be 'variables'
283+
# For now much match the trial parameter names.
284+
point["ax_trial_id"] = trial_index
285+
point["arm_name"] = arm.name
286+
point["trial_type"] = trial_type
287+
points.append(point)
288288
return points
289289

290290
def ingest(self, results: List[dict]) -> None:

0 commit comments

Comments
 (0)