Minor fix

zhiqwang · Dec 20, 2021 · 87a412e · 87a412e
1 parent b2b7f06
commit 87a412e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test/test_models_anchor_utils.py b/test/test_models_anchor_utils.py
@@ -26,5 +26,5 @@ def test_anchor_generator(self):
         assert tuple(anchors[0][0].shape) == (1, 1, 2, 2, 2)
         assert tuple(anchors[1][0].shape) == (1, 1, 2, 2, 2)
 
-        torch.testing.assert_close(anchors[0][0], expected_grids, rtol=0, atol=0)
-        torch.testing.assert_close(anchors[1][0], expected_shifts, rtol=0, atol=0)
+        torch.testing.assert_close(anchors[0][0], expected_grids)
+        torch.testing.assert_close(anchors[1][0], expected_shifts)