fix

huggingface · Aug 5, 2024 · d71342b · d71342b
1 parent 1f14e36
commit d71342b
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/tests/openvino/test_diffusion.py b/tests/openvino/test_diffusion.py
@@ -248,7 +248,7 @@ def test_compare_to_diffusers(self, model_arch: str):
             self.assertTrue(np.allclose(ov_outputs, outputs, atol=1e-4))
 
         # Compare model devices
-        self.assertEqual(pipeline.device.type, ov_pipeline.device)
+        self.assertEqual(pipeline.device, ov_pipeline.device)
 
     @parameterized.expand(SUPPORTED_ARCHITECTURES)
     def test_image_reproducibility(self, model_arch: str):
@@ -406,7 +406,7 @@ def test_compare_to_diffusers(self, model_arch: str):
             # Compare model outputs
             self.assertTrue(np.allclose(ov_outputs, outputs, atol=1e-4))
         # Compare model devices
-        self.assertEqual(pipeline.device.type, ov_pipeline.device)
+        self.assertEqual(pipeline.device, ov_pipeline.device)
 
     @parameterized.expand(SUPPORTED_ARCHITECTURES)
     def test_image_reproducibility(self, model_arch: str):
@@ -536,7 +536,7 @@ def test_compare_to_diffusers(self, model_arch: str):
             # Compare model outputs
             self.assertTrue(np.allclose(ov_outputs, outputs, atol=1e-4))
         # Compare model devices
-        self.assertEqual(pipeline.device.type, ov_pipeline.device)
+        self.assertEqual(pipeline.device, ov_pipeline.device)
 
     @parameterized.expand(SUPPORTED_ARCHITECTURES)
     @pytest.mark.run_slow
@@ -602,4 +602,4 @@ def test_safety_checker(self, model_arch: str):
             # Compare model outputs
             self.assertTrue(np.allclose(ov_outputs, outputs, atol=1e-4))
         # Compare model devices
-        self.assertEqual(pipeline.device.type, ov_pipeline.device)
+        self.assertEqual(pipeline.device, ov_pipeline.device)