Try a more tolerant check

openmm · Aug 23, 2024 · b2a7970 · b2a7970
1 parent efaf8de
commit b2a7970
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/python/tests/TestTorchForce.py b/python/tests/TestTorchForce.py
@@ -87,9 +87,7 @@ def forward(self, positions):
  assert self.positions.device == self.device
  assert positions.device == self.device
  assert positions.dtype == self.dtype
- print(positions)
- print(self.positions)
- assert pt.all(positions == self.positions)
+ assert pt.allclose(positions, self.positions)
  return pt.sum(positions)
 
  with NamedTemporaryFile() as fd: