diff --git a/tests/openvino/test_export.py b/tests/openvino/test_export.py index 0d027dffc..80a020d2b 100644 --- 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):