From 7f3b7cf6d52f14e191979cb3d96b55a616afae59 Mon Sep 17 00:00:00 2001 From: Alexander Date: Fri, 15 Dec 2023 13:40:19 +0400 Subject: [PATCH] Style --- optimum/exporters/openvino/__main__.py | 7 ++++++- 1 file changed, 6 insertions(+), 1 deletion(-) diff --git a/optimum/exporters/openvino/__main__.py b/optimum/exporters/openvino/__main__.py index 30f0c17038..542199fa4f 100644 --- 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`" )