Skip to content

Commit 878bb1d

Browse files
committed
Ruff format and lint
stack-info: PR: #1226, branch: drisspg/stack/17
1 parent 6fd77d5 commit 878bb1d

File tree

10 files changed

+878
-314
lines changed

10 files changed

+878
-314
lines changed

ruff.toml

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -11,6 +11,6 @@ include = [
1111
"torchao/quantization/linear_activation_weight_observer.py",
1212
"test/quantization/test_observer.py",
1313
"test/dtypes/test_affine_quantized_float.py",
14-
"torchao/quantization/weight_tensor_linear_activation_quantization.py"
15-
14+
"torchao/quantization/weight_tensor_linear_activation_quantization.py",
15+
"torchao/dtypes/**/*.py",
1616
]

torchao/dtypes/__init__.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
from .nf4tensor import NF4Tensor, to_nf4
2+
23
# from ..prototype.dtypes.uint2 import UInt2Tensor, BitnetTensor
34
from .uint4 import UInt4Tensor
45
from .affine_quantized_tensor import (
@@ -21,7 +22,7 @@
2122
__all__ = [
2223
"NF4Tensor",
2324
"to_nf4",
24-
"UInt4Tensor"
25+
"UInt4Tensor",
2526
"AffineQuantizedTensor",
2627
"to_affine_quantized_intx",
2728
"to_affine_quantized_intx_static",

0 commit comments

Comments
 (0)