formatting

Author: shumaari

nncf/common/graph/utils.py

@@ -157,11 +157,13 @@ def get_weight_shape_legacy(layer_attributes: WeightedLayerAttributes) -> List[i
             return [
                 layer_attributes.out_channels,
                 layer_attributes.in_channels // layer_attributes.groups,
-                *layer_attributes.kernel_size]
+                *layer_attributes.kernel_size,
+            ]
         return [
             layer_attributes.in_channels,
             layer_attributes.out_channels // layer_attributes.groups,
-            *layer_attributes.kernel_size]
+            *layer_attributes.kernel_size,
+        ]
 
     if isinstance(layer_attributes, GroupNormLayerAttributes):
         return [layer_attributes.num_channels]
@@ -204,4 +206,3 @@ def get_num_filters_legacy(layer_attributes: WeightedLayerAttributes) -> int:
     """
     weight_shape = layer_attributes.get_weight_shape_legacy()
     return weight_shape[layer_attributes.get_target_dim_for_compression_legacy()]
-    

nncf/experimental/torch/sparsity/movement/layers.py

@@ -18,8 +18,8 @@
 
 import nncf
 from nncf.common.graph import NNCFNode
-from nncf.common.graph.utils import get_weight_shape_legacy
 from nncf.common.graph.utils import get_bias_shape_legacy
+from nncf.common.graph.utils import get_weight_shape_legacy
 from nncf.experimental.torch.sparsity.movement.functions import binary_mask_by_threshold
 from nncf.torch.layer_utils import COMPRESSION_MODULES
 from nncf.torch.layer_utils import CompressionParameter

nncf/experimental/torch/sparsity/movement/structured_mask_handler.py

@@ -18,8 +18,8 @@
 
 from nncf.common.graph.graph import NNCFNodeName
 from nncf.common.graph.layer_attributes import LinearLayerAttributes
-from nncf.common.graph.utils import get_weight_shape_legacy
 from nncf.common.graph.utils import get_bias_shape_legacy
+from nncf.common.graph.utils import get_weight_shape_legacy
 from nncf.common.logging import nncf_logger
 from nncf.experimental.common.pruning.nodes_grouping import get_pruning_groups
 from nncf.experimental.common.pruning.nodes_grouping import select_largest_groups

nncf/torch/quantization/algo.py

@@ -35,8 +35,8 @@
 from nncf.common.graph.patterns.manager import TargetDevice
 from nncf.common.graph.transformations.commands import TargetType
 from nncf.common.graph.utils import get_first_nodes_of_type
-from nncf.common.graph.utils import get_weight_shape_legacy
 from nncf.common.graph.utils import get_target_dim_for_compression_legacy
+from nncf.common.graph.utils import get_weight_shape_legacy
 from nncf.common.hardware.config import HWConfig
 from nncf.common.hardware.config import HWConfigType
 from nncf.common.hardware.config import get_hw_config_type

nncf/torch/quantization/init_range.py

@@ -18,8 +18,8 @@
 
 import nncf
 from nncf.common.graph.layer_attributes import WeightedLayerAttributes
-from nncf.common.graph.utils import get_weight_shape_legacy
 from nncf.common.graph.utils import get_target_dim_for_compression_legacy
+from nncf.common.graph.utils import get_weight_shape_legacy
 from nncf.common.quantization.initialization.range import RangeInitCollectorParams
 from nncf.common.quantization.initialization.range import RangeInitConfig
 from nncf.common.quantization.initialization.range import RangeInitParams