Fix style

huggingface · Jan 19, 2024 · 59a1e15 · 59a1e15
1 parent c886e69
commit 59a1e15
Showing 1 changed file with 9 additions and 2 deletions.
diff --git a/optimum/exporters/openvino/convert.py b/optimum/exporters/openvino/convert.py
@@ -31,7 +31,14 @@
 from optimum.exporters.onnx.model_patcher import DecoderModelPatcher
 from optimum.utils import is_diffusers_available
 
-from ...intel.utils.import_utils import _torch_version, _transformers_version, is_nncf_available, is_optimum_version, is_transformers_version
+from ...intel.utils.import_utils import (
+    _torch_version,
+    _transformers_version,
+    is_nncf_available,
+    is_optimum_version,
+    is_torch_version,
+    is_transformers_version,
+)
 from .model_patcher import patch_model_with_bettertransformer
 from .stateful import ensure_stateful_is_available, patch_stateful
 from .utils import (
@@ -323,7 +330,7 @@ def export_pytorch(
     output = Path(output)
 
     if stateful:
-        if is_transformers_version("<", "4.36") or  is_torch_version("<", "2.1.1"):
+        if is_transformers_version("<", "4.36") or is_torch_version("<", "2.1.1"):
             COLOR_RED = "\033[1;31m"
             COLOR_RESET = "\033[0m"
             logger.warning(