mitigate issue

Signed-off-by: Yiheng Wang <[email protected]>
Project-MONAI · Dec 30, 2024 · 0a8dfea · 0a8dfea
1 parent d36f0c8
commit 0a8dfea
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_load_image.py b/tests/test_load_image.py
@@ -233,7 +233,7 @@ def test_nibabel_reader_gpu(self, input_param, filenames, expected_shape):
             input_param_cpu = input_param.copy()
             input_param_cpu["to_gpu"] = False
             result_cpu = LoadImage(image_only=True, **input_param_cpu)(filenames)
-            self.assertTrue(torch.equal(result_cpu, result.cpu()))
+            self.assertTrue(torch.allclose(result_cpu, result.cpu(), atol=1e-6))
 
     @parameterized.expand([TEST_CASE_6, TEST_CASE_7, TEST_CASE_8, TEST_CASE_8_1, TEST_CASE_9])
     def test_itk_reader(self, input_param, filenames, expected_shape):