fix style post rename PR (vllm-project#25)

src/compressed_tensors/compressors/sparse_bitmask.py

@@ -17,9 +17,9 @@
 
 import numpy
 import torch
-from safetensors import safe_open
 from compressed_tensors.compressors import ModelCompressor
 from compressed_tensors.utils import get_nested_weight_mappings, merge_names
+from safetensors import safe_open
 from torch import Tensor
 from tqdm import tqdm
 

src/compressed_tensors/config/base.py

@@ -14,8 +14,8 @@
 
 from typing import Optional
 
-from pydantic import BaseModel
 from compressed_tensors.registry import RegistryMixin
+from pydantic import BaseModel
 
 
 __all__ = ["CompressionConfig"]

src/compressed_tensors/quantization/lifecycle/apply.py

@@ -16,7 +16,9 @@
 from collections import OrderedDict
 from typing import Iterable, Optional
 
-from compressed_tensors.quantization.lifecycle.calibration import set_module_for_calibration
+from compressed_tensors.quantization.lifecycle.calibration import (
+    set_module_for_calibration,
+)
 from compressed_tensors.quantization.lifecycle.frozen import freeze_module_quantization
 from compressed_tensors.quantization.lifecycle.initialize import (
     initialize_module_for_quantization,

src/compressed_tensors/quantization/lifecycle/initialize.py

@@ -17,7 +17,9 @@
 from typing import Optional
 
 import torch
-from compressed_tensors.quantization.lifecycle.forward import wrap_module_forward_quantized
+from compressed_tensors.quantization.lifecycle.forward import (
+    wrap_module_forward_quantized,
+)
 from compressed_tensors.quantization.quant_args import QuantizationArgs
 from compressed_tensors.quantization.quant_config import QuantizationStatus
 from compressed_tensors.quantization.quant_scheme import QuantizationScheme

src/compressed_tensors/quantization/quant_config.py

@@ -15,14 +15,14 @@
 from enum import Enum
 from typing import Dict, List, Optional
 
-from pydantic import BaseModel, Field
 from compressed_tensors.quantization.quant_scheme import QuantizationScheme
 from compressed_tensors.quantization.utils import (
     calculate_compression_ratio,
     is_module_quantized,
     iter_named_leaf_modules,
     module_type,
 )
+from pydantic import BaseModel, Field
 from torch.nn import Module
 
 

src/compressed_tensors/quantization/quant_scheme.py

@@ -14,8 +14,8 @@
 
 from typing import List, Optional
 
-from pydantic import BaseModel
 from compressed_tensors.quantization.quant_args import QuantizationArgs
+from pydantic import BaseModel
 
 
 __all__ = ["QuantizationScheme"]

tests/quantization/test_quant_args.py

@@ -13,12 +13,12 @@
 # limitations under the License.
 
 import pytest
-from pydantic import ValidationError
 from compressed_tensors.quantization import (
     QuantizationArgs,
     QuantizationStrategy,
     QuantizationType,
 )
+from pydantic import ValidationError
 
 
 def test_defaults():

tests/quantization/test_quant_config.py

@@ -14,12 +14,12 @@
 
 
 import pytest
-from pydantic import ValidationError
 from compressed_tensors.quantization import (
     QuantizationConfig,
     QuantizationScheme,
     QuantizationStatus,
 )
+from pydantic import ValidationError
 
 
 def test_basic_config():

tests/quantization/test_quant_scheme.py

@@ -13,8 +13,8 @@
 # limitations under the License.
 
 import pytest
-from pydantic import ValidationError
 from compressed_tensors.quantization import QuantizationArgs, QuantizationScheme
+from pydantic import ValidationError
 
 
 def test_basic_scheme():

tests/test_bitmask.py

@@ -17,8 +17,8 @@
 
 import pytest
 import torch
-from safetensors.torch import save_file
 from compressed_tensors import BitmaskCompressor, BitmaskConfig, BitmaskTensor
+from safetensors.torch import save_file
 
 
 @pytest.mark.parametrize(