formatting

Signed-off-by: Brian Dellabetta <[email protected]>

Author: brian-dellabetta

tests/llmcompressor/pytorch/utils/test_sparse.py

@@ -1,6 +1,11 @@
 import pytest
 import torch
-from compressed_tensors.quantization import QuantizationArgs, QuantizationScheme, QuantizationStrategy, QuantizationType
+from compressed_tensors.quantization import (
+    QuantizationArgs,
+    QuantizationScheme,
+    QuantizationStrategy,
+    QuantizationType,
+)
 from torch.nn import Linear, Module, ReLU
 
 from llmcompressor.pytorch.utils import ModuleSparsificationInfo
@@ -14,16 +19,16 @@ def __init__(self):
         self.relu = ReLU()
 
         self.fc1.quantization_scheme = QuantizationScheme(
-                targets=["Linear"],
-                weights=QuantizationArgs(
-                    num_bits=4,
-                    type=QuantizationType.INT,
-                    group_size=128,
-                    strategy=QuantizationStrategy.GROUP,
-                    symmetric=True,
-                    dynamic=False,
-                )
-            )
+            targets=["Linear"],
+            weights=QuantizationArgs(
+                num_bits=4,
+                type=QuantizationType.INT,
+                group_size=128,
+                strategy=QuantizationStrategy.GROUP,
+                symmetric=True,
+                dynamic=False,
+            ),
+        )
 
 
 def test_module_quantization_info():