Apply automatic Ruff fixes

pytorch · Nov 21, 2024 · b9d87f5 · b9d87f5
1 parent aa7bac0
commit b9d87f5
Show file tree

Hide file tree

Showing 4 changed files with 10 additions and 5 deletions.
diff --git a/torchao/dtypes/__init__.py b/torchao/dtypes/__init__.py
@@ -16,12 +16,12 @@
 from .nf4tensor import NF4Tensor, to_nf4
 from .uintx import (
     BlockSparseLayout,
+    Int4CPULayout,
     MarlinQQQLayout,
     MarlinSparseLayout,
     SemiSparseLayout,
     TensorCoreTiledLayout,
     UintxLayout,
-    Int4CPULayout,
 )
 from .utils import (
     Layout,

diff --git a/torchao/dtypes/uintx/__init__.py b/torchao/dtypes/uintx/__init__.py
@@ -11,8 +11,8 @@
     SemiSparseLayout,
 )
 from .tensor_core_tiled_layout import (
-    TensorCoreTiledLayout,
     Int4CPULayout,
+    TensorCoreTiledLayout,
 )
 from .uintx_layout import (
     UintxLayout,

diff --git a/torchao/dtypes/uintx/tensor_core_tiled_layout.py b/torchao/dtypes/uintx/tensor_core_tiled_layout.py
@@ -13,7 +13,12 @@
 )
 from torchao.dtypes.utils import AQTTensorImpl, Layout, is_device
 from torchao.quantization.quant_primitives import ZeroPointDomain, _get_reduction_params
-from torchao.utils import TORCH_VERSION_AT_LEAST_2_5, TORCH_VERSION_AT_LEAST_2_6, fill_defaults, find_multiple
+from torchao.utils import (
+    TORCH_VERSION_AT_LEAST_2_5,
+    TORCH_VERSION_AT_LEAST_2_6,
+    fill_defaults,
+    find_multiple,
+)
 
 aten = torch.ops.aten
 

diff --git a/torchao/quantization/GPTQ.py b/torchao/quantization/GPTQ.py
@@ -616,8 +616,8 @@ def __init__(
                     ),
                     dtype=torch.uint8,
                     device=device,
-            ),
-        )
+                ),
+            )
         else:
             self.register_buffer(
                 "weight",