text/style fixes

Signed-off-by: Brian Dellabetta <bdellabe@redhat.com>

Author: brian-dellabetta

src/compressed_tensors/compressors/model_compressors/model_compressor.py

@@ -338,10 +338,10 @@ def __init__(
 
             self.quantization_compressor = {}
             for format in self.compression_formats:
-                self.quantization_compressor[
-                    format
-                ] = BaseCompressor.load_from_registry(
-                    format, config=quantization_config
+                self.quantization_compressor[format] = (
+                    BaseCompressor.load_from_registry(
+                        format, config=quantization_config
+                    )
                 )
 
     def get_missing_module_keys(self, model: Module) -> List[str]:

src/compressed_tensors/quantization/utils/helpers.py

@@ -115,9 +115,11 @@ def calculate_qparams(
     # 4. Update any 0s with small values to
     # prevent div by 0
     eps = _get_dtype_eps(
-        dtype=quantization_args.scale_dtype
-        if quantization_args.scale_dtype is not None
-        else scales.dtype
+        dtype=(
+            quantization_args.scale_dtype
+            if quantization_args.scale_dtype is not None
+            else scales.dtype
+        )
     )
     scales = torch.where(
         scales == 0,

src/compressed_tensors/utils/helpers.py

@@ -16,7 +16,17 @@
 import warnings
 from functools import wraps
 from types import MappingProxyType
-from typing import TYPE_CHECKING, Any, Callable, Dict, List, Mapping, Optional, TypeVar
+from typing import (
+    TYPE_CHECKING,
+    Any,
+    Callable,
+    Dict,
+    Iterable,
+    List,
+    Mapping,
+    Optional,
+    TypeVar,
+)
 
 import numpy
 import torch
@@ -391,9 +401,9 @@ def patch_attrs(bases: Iterable[Any], attr: str, values: Iterable[Any]):
     >>> assert not hasattr(obj1, "attribute")
     >>> assert not hasattr(obj2, "attribute")
     """
-    with contextlib.exitstack() as stack:
+    with contextlib.ExitStack() as stack:
         for base, value in zip(bases, values):
-            stack.add(patch_attr(base, attr, value))
+            stack.enter_context(patch_attr(base, attr, value))
         yield