update config

Signed-off-by: Sun, Xuehao <[email protected]>

Author: XuehaoSun

.pre-commit-config.yaml

@@ -88,6 +88,11 @@ repos:
     hooks:
       - id: black
         files: (.*\.py)$
+        exclude: |
+          (?x)^(
+              neural_compressor/conf/config.py|
+              neural_compressor/conf/pythonic_config.py
+          )$
 
   - repo: https://github.com/asottile/blacken-docs
     rev: 1.16.0

neural_compressor/adaptor/tf_utils/quantize_graph/qdq/fuse_qdq_conv.py

@@ -37,6 +37,7 @@ def __init__(self, **kwargs):
                                       key=lambda i: len(i),
                                       reverse=True)
         if self.new_api:
+            # fmt: off
             self.fusion_mapping = {
                 'DequantizeConv2DBiasAddQuantizeV2': self.apply_newly_conv_biasadd_fusion,
                 'DequantizeConv2DBiasAddAddNReluQuantizeV2': self.apply_newly_conv_biasadd_addn_relu_fusion,
@@ -127,6 +128,7 @@ def __init__(self, **kwargs):
                 'DequantizeConv3DAddLeakyReluQuantizeV2': self.apply_conv3d_add_relu_fusion,
                 'DequantizeDepthwiseConv2dNativeQuantizeV2': self.apply_newly_conv_single_fusion
             }
+            # fmt: on
 
     def _insert_dummy_biasadd(self, match_node_name, matched_node):
         """Insert dummy biasadd for fusion."""

neural_compressor/experimental/data/transforms/transform.py

@@ -2680,12 +2680,14 @@ class ParseDecodeVocTransform(BaseTransform):
         tuple of parsed image and labels
     """
 
+    # fmt: off
     def __call__(self, sample):
         """Parse decode voc."""
         # Currently only supports jpeg and png.
         # Need to use this logic because the shape is not known for
         # tf.image.decode_image and we rely on this info to
         # extend label if necessary.
+        # fmt: on
         def _decode_image(content, channels):
             """Decode the image with content."""
             return tf.cond(

pyproject.toml

@@ -29,6 +29,7 @@ ignore = [
     "E402", # Module level import not at top of file
     "E501", # Line too long (121 > 120 characters)
     "E721", # Do not compare types, use isinstance()
+    "E731", # Do not assign a lambda expression, use a def
     "E722", # Do not use bare except
     "E741", # Do not use variables named ‘l’, ‘O’, or ‘I’
     "F401", # {name} imported but unused