Merge branch 'main' into export-D54879641

facebook · Mar 18, 2024 · 70a3605 · 70a3605
2 parents 78cb5eb + 23636ed
commit 70a3605
Show file tree

Hide file tree

Showing 3 changed files with 402 additions and 1 deletion.
diff --git a/ax/modelbridge/generation_strategy.py b/ax/modelbridge/generation_strategy.py
@@ -386,7 +386,7 @@ def _validate_and_set_step_sequence(self, steps: List[GenerationStep]) -> None:
 
             # Set transition_to field for all but the last step, which remains
             # null.
-            if idx != len(self._steps):
+            if idx != len(self._steps) - 1:
                 for transition_criteria in step.transition_criteria:
                     if (
                         transition_criteria.criterion_class