fix tests

bdaiinstitute · Aug 23, 2024 · c6f3718 · c6f3718
1 parent 5e2ac3f
commit c6f3718
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/predicators/structs.py b/predicators/structs.py
@@ -185,6 +185,8 @@ def allclose(self, other: State) -> bool:
             if not CFG.allow_state_allclose_comparison_despite_simulator_state:
                 raise NotImplementedError("Cannot use allclose when "
                                           "simulator_state is not None.")
+            if self.simulator_state != other.simulator_state:
+                return False
         return self._allclose(other)
 
     def _allclose(self, other: State) -> bool: