Ruff format and lint

stack-info: PR: #1226, branch: drisspg/stack/17
pytorch · Nov 5, 2024 · 878bb1d · 878bb1d
1 parent 6fd77d5
commit 878bb1d
Show file tree

Hide file tree

Showing 10 changed files with 878 additions and 314 deletions.
diff --git a/ruff.toml b/ruff.toml
@@ -11,6 +11,6 @@ include = [
     "torchao/quantization/linear_activation_weight_observer.py",
     "test/quantization/test_observer.py",
     "test/dtypes/test_affine_quantized_float.py",
-    "torchao/quantization/weight_tensor_linear_activation_quantization.py"
-
+    "torchao/quantization/weight_tensor_linear_activation_quantization.py",
+    "torchao/dtypes/**/*.py",
 ]
diff --git a/torchao/dtypes/__init__.py b/torchao/dtypes/__init__.py
@@ -1,4 +1,5 @@
 from .nf4tensor import NF4Tensor, to_nf4
+
 # from ..prototype.dtypes.uint2 import UInt2Tensor, BitnetTensor
 from .uint4 import UInt4Tensor
 from .affine_quantized_tensor import (
@@ -21,7 +22,7 @@
 __all__ = [
     "NF4Tensor",
     "to_nf4",
-    "UInt4Tensor"
+    "UInt4Tensor",
     "AffineQuantizedTensor",
     "to_affine_quantized_intx",
     "to_affine_quantized_intx_static",