Style

huggingface · Dec 15, 2023 · 7f3b7cf · 7f3b7cf
1 parent d878453
commit 7f3b7cf
Showing 1 changed file with 6 additions and 1 deletion.
diff --git a/optimum/exporters/openvino/__main__.py b/optimum/exporters/openvino/__main__.py
@@ -134,7 +134,12 @@ def main_export(
     >>> main_export("gpt2", output="gpt2_onnx/")
     ```
     """
-    if compression_option is not None and compression_option != "f16" and compression_option != "f32" and not is_nncf_available():
+    if (
+        compression_option is not None
+        and compression_option != "f16"
+        and compression_option != "f32"
+        and not is_nncf_available()
+    ):
         raise ImportError(
             f"Compression of the weights to {compression_option} requires nncf, please install it with `pip install nncf`"
         )