Tests: replace torch.testing.assert_allclose by torch.testing.assert_close (#29915)

* replace torch.testing.assert_allclose by torch.testing.assert_close * missing atol rtol
2024-03-28 09:53:31 +00:00
parent 7c19fafe44
commit 248d5d23a2
7 changed files with 30 additions and 34 deletions
--- a/tests/models/umt5/test_modeling_umt5.py
+++ b/tests/models/umt5/test_modeling_umt5.py
@@ -763,7 +763,7 @@ class Umt5IntegrationTest(unittest.TestCase):
            ]
        )
        # fmt: on
-        torch.testing.assert_allclose(input_ids, EXPECTED_IDS)
+        torch.testing.assert_close(input_ids, EXPECTED_IDS)

        generated_ids = model.generate(input_ids.to(torch_device))
        EXPECTED_FILLING = [