Update rescale tests - cast to float after rescaling to reflect huggi…

…ngface#25229 (huggingface#25259) Rescale tests - cast to float after rescaling to reflect huggingface#25229
blbadger · Nov 8, 2023 · 3489d22 · 3489d22
1 parent 9ad2ab6
commit 3489d22
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/tests/models/efficientnet/test_image_processing_efficientnet.py b/tests/models/efficientnet/test_image_processing_efficientnet.py
@@ -201,9 +201,9 @@ def test_rescale(self):
         image_processor = self.image_processing_class(**self.image_processor_dict)
 
         rescaled_image = image_processor.rescale(image, scale=1 / 255)
-        expected_image = image.astype(np.float32) * (2 / 255.0) - 1
+        expected_image = (image * (2 / 255.0)).astype(np.float32) - 1
         self.assertTrue(np.allclose(rescaled_image, expected_image))
 
         rescaled_image = image_processor.rescale(image, scale=1 / 255, offset=False)
-        expected_image = image.astype(np.float32) / 255.0
+        expected_image = (image / 255.0).astype(np.float32)
         self.assertTrue(np.allclose(rescaled_image, expected_image))
diff --git a/tests/models/vivit/test_image_processing_vivit.py b/tests/models/vivit/test_image_processing_vivit.py
@@ -220,9 +220,9 @@ def test_rescale(self):
         image_processor = self.image_processing_class(**self.image_processor_dict)
 
         rescaled_image = image_processor.rescale(image, scale=1 / 255)
-        expected_image = image.astype(np.float32) * (2 / 255.0) - 1
+        expected_image = (image * (2 / 255.0)).astype(np.float32) - 1
         self.assertTrue(np.allclose(rescaled_image, expected_image))
 
         rescaled_image = image_processor.rescale(image, scale=1 / 255, offset=False)
-        expected_image = image.astype(np.float32) / 255.0
+        expected_image = (image / 255.0).astype(np.float32)
         self.assertTrue(np.allclose(rescaled_image, expected_image))