format

Author: echarlaix

optimum/intel/__init__.py

@@ -60,11 +60,23 @@
         raise OptionalDependencyNotAvailable()
 except OptionalDependencyNotAvailable:
     _import_structure["utils.dummy_openvino_and_nncf_objects"].extend(
-        ["OVQuantizer", "OVTrainingArguments", "OVQuantizationConfig", "OVWeightQuantizationConfig", "OVDynamicQuantizationConfig"]
+        [
+            "OVQuantizer",
+            "OVTrainingArguments",
+            "OVQuantizationConfig",
+            "OVWeightQuantizationConfig",
+            "OVDynamicQuantizationConfig",
+        ]
     )
 else:
     _import_structure["openvino"].extend(
-        ["OVQuantizer", "OVTrainingArguments", "OVQuantizationConfig", "OVWeightQuantizationConfig", "OVDynamicQuantizationConfig"]
+        [
+            "OVQuantizer",
+            "OVTrainingArguments",
+            "OVQuantizationConfig",
+            "OVWeightQuantizationConfig",
+            "OVDynamicQuantizationConfig",
+        ]
     )
 
 
@@ -191,14 +203,20 @@
             raise OptionalDependencyNotAvailable()
     except OptionalDependencyNotAvailable:
         from .utils.dummy_openvino_and_nncf_objects import (
+            OVDynamicQuantizationConfig,
             OVQuantizationConfig,
             OVQuantizer,
             OVTrainingArguments,
             OVWeightQuantizationConfig,
-            OVDynamicQuantizationConfig,
         )
     else:
-        from .openvino import OVQuantizationConfig, OVQuantizer, OVTrainingArguments, OVWeightQuantizationConfig, OVDynamicQuantizationConfig
+        from .openvino import (
+            OVDynamicQuantizationConfig,
+            OVQuantizationConfig,
+            OVQuantizer,
+            OVTrainingArguments,
+            OVWeightQuantizationConfig,
+        )
 
     try:
         if not (is_openvino_available() and is_nncf_available() and is_accelerate_available()):

optimum/intel/openvino/__init__.py

@@ -43,7 +43,7 @@
         from .trainer import OVTrainer
 
 
-from .configuration import OVConfig, OVQuantizationConfig, OVWeightQuantizationConfig, OVDynamicQuantizationConfig
+from .configuration import OVConfig, OVDynamicQuantizationConfig, OVQuantizationConfig, OVWeightQuantizationConfig
 from .modeling import (
     OVModelForAudioClassification,
     OVModelForAudioFrameClassification,

tests/openvino/test_quantization.py

@@ -685,7 +685,7 @@ class OVQuantizationConfigTest(unittest.TestCase):
     QUANTIZATION_CONFIGS = (
         (None,),
         (OVWeightQuantizationConfig(),),
-        (OVWeightQuantizationConfig(bits=8,sym=True),),
+        (OVWeightQuantizationConfig(bits=8, sym=True),),
         (
             OVWeightQuantizationConfig(
                 dataset="wikitext",
@@ -713,7 +713,7 @@ class OVQuantizationConfigTest(unittest.TestCase):
             ),
         ),
         (OVQuantizationConfig(ignored_scope=nncf.IgnoredScope(names=["op_name"])),),
-        (OVDynamicQuantizationConfig(bits=8,sym=True),),
+        (OVDynamicQuantizationConfig(bits=8, sym=True),),
     )
 
     QUANTIZATION_CONFIG_DICTS = (
@@ -743,7 +743,11 @@ class OVQuantizationConfigTest(unittest.TestCase):
         ),
         (dict(num_samples=100), OVWeightQuantizationConfig, "Can't determine type of OV quantization config"),
         (dict(abc="def"), OVWeightQuantizationConfig, "Can't determine type of OV quantization config"),
-        (dict(bits=8, fast_bias_correction=True, dataset="wikitext"), OVWeightQuantizationConfig, "Can't determine type of OV quantization config"),
+        (
+            dict(bits=8, fast_bias_correction=True, dataset="wikitext"),
+            OVWeightQuantizationConfig,
+            "Can't determine type of OV quantization config",
+        ),
         (dict(model_type="transformer"), OVQuantizationConfig, None),
         (
             dict(
@@ -761,7 +765,11 @@ class OVQuantizationConfigTest(unittest.TestCase):
         (dict(weight_only=False), OVQuantizationConfig, None),
         (dict(abc="def", weight_only=False), OVQuantizationConfig, None),
         (dict(abc="def", weight_only=True), OVWeightQuantizationConfig, None),
-        (dict(bits=8, fast_bias_correction=True, dataset="wikitext", weight_only=True), OVWeightQuantizationConfig, None),
+        (
+            dict(bits=8, fast_bias_correction=True, dataset="wikitext", weight_only=True),
+            OVWeightQuantizationConfig,
+            None,
+        ),
         (dict(bits=8, fast_bias_correction=True, weight_only=False), OVQuantizationConfig, None),
     )