weird rebase behavior

determined-ai · Mar 28, 2024 · dd6a59c · dd6a59c
1 parent bf8d210
commit dd6a59c
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/harness/determined/pytorch/_pytorch_trial.py b/harness/determined/pytorch/_pytorch_trial.py
@@ -104,8 +104,8 @@ def _from_values(
     def should_stop(self, step_num: int) -> bool:
         if isinstance(self.value, int):
             return self._divides(step_num)
-        if isinstance(self.value, col_abc.Container):
-            return step_num in self.value
+        assert isinstance(self.value, col_abc.Container)
+        return step_num in self.value
 
     def _divides(self, steps: int) -> bool:
         assert isinstance(steps, int) and isinstance(