From 9474abdf4735d8c18ffeb359202b7a4e19452cf9 Mon Sep 17 00:00:00 2001 From: Yih-Dar <2521628+ydshieh@users.noreply.github.com> Date: Mon, 6 Mar 2023 17:41:00 +0100 Subject: [PATCH] Use larger atol in `torch.allclose` for some tests (#21966) Use larger atol Co-authored-by: ydshieh --- tests/test_modeling_common.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/test_modeling_common.py b/tests/test_modeling_common.py index 816a959a4e..5cb04edbb1 100755 --- a/tests/test_modeling_common.py +++ b/tests/test_modeling_common.py @@ -2497,7 +2497,7 @@ class ModelTesterMixin: torch.manual_seed(0) new_output = new_model(**inputs_dict_class) - self.assertTrue(torch.allclose(base_output[0], new_output[0])) + self.assertTrue(torch.allclose(base_output[0], new_output[0], atol=1e-5)) @require_accelerate @mark.accelerate_tests @@ -2533,7 +2533,7 @@ class ModelTesterMixin: torch.manual_seed(0) new_output = new_model(**inputs_dict_class) - self.assertTrue(torch.allclose(base_output[0], new_output[0])) + self.assertTrue(torch.allclose(base_output[0], new_output[0], atol=1e-5)) @require_accelerate @mark.accelerate_tests @@ -2569,7 +2569,7 @@ class ModelTesterMixin: torch.manual_seed(0) new_output = new_model(**inputs_dict_class) - self.assertTrue(torch.allclose(base_output[0], new_output[0])) + self.assertTrue(torch.allclose(base_output[0], new_output[0], atol=1e-5)) def test_problem_types(self): config, inputs_dict = self.model_tester.prepare_config_and_inputs_for_common()