Style

huggingface · Nov 11, 2024 · 14ed02c · 14ed02c
1 parent efb98bc
commit 14ed02c
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/tests/openvino/test_export.py b/tests/openvino/test_export.py
@@ -127,8 +127,12 @@ def _openvino_export(
                     self.assertTrue(ov_model.model.has_rt_info(["runtime_options", "ACTIVATIONS_SCALE_FACTOR"]))
 
                 if library_name == "diffusers":
-                    self.assertTrue(ov_model.vae_encoder.model.has_rt_info(["runtime_options", "ACTIVATIONS_SCALE_FACTOR"]))
-                    self.assertTrue(ov_model.vae_decoder.model.has_rt_info(["runtime_options", "ACTIVATIONS_SCALE_FACTOR"]))
+                    self.assertTrue(
+                        ov_model.vae_encoder.model.has_rt_info(["runtime_options", "ACTIVATIONS_SCALE_FACTOR"])
+                    )
+                    self.assertTrue(
+                        ov_model.vae_decoder.model.has_rt_info(["runtime_options", "ACTIVATIONS_SCALE_FACTOR"])
+                    )
 
     @parameterized.expand(SUPPORTED_ARCHITECTURES)
     def test_export(self, model_type: str):