Lint fixes test/quantization

ruff.toml

@@ -9,7 +9,7 @@ include = [
     "torchao/sparsity/**/*.py",
     "torchao/prototype/low_bit_optim/**.py",
     "test/float8/**/*.py",
-    "test/quantization/test_observer.py",
+    "test/quantization/**/*.py",
     "test/dtypes/**/*.py",
     "test/prototype/low_bit_optim/**.py",
     "torchao/utils.py",

test/quantization/test_galore_quant.py

@@ -3,13 +3,16 @@
 import pytest
 
 # Skip entire test if triton is not available, otherwise CI failure
-try:
-    import triton
-except ImportError:
-    pytest.skip("triton is not installed", allow_module_level=True)
-
-from bitsandbytes.functional import create_dynamic_map, quantize_blockwise, dequantize_blockwise
+try:  # noqa: F401
+    import triton  # noqa: F401
+except ImportError:  # noqa: F401
+    pytest.skip("triton is not installed", allow_module_level=True)  # noqa: F401
 import torch
+from bitsandbytes.functional import (
+    create_dynamic_map,
+    dequantize_blockwise,
+    quantize_blockwise,
+)
 
 from torchao.prototype.galore.kernels import (
     triton_dequant_blockwise,

test/quantization/test_marlin_qqq.py

@@ -1,4 +1,5 @@
 import copy
+import unittest
 
 import pytest
 import torch
@@ -19,9 +20,12 @@
     choose_qparams_and_quantize_affine_qqq,
 )
 from torchao.utils import TORCH_VERSION_AT_LEAST_2_5, is_fbcode
-import unittest
 
-@unittest.skipIf(is_fbcode(), "Skipping the test in fbcode since we don't have TARGET file for kernels")
+
+@unittest.skipIf(
+    is_fbcode(),
+    "Skipping the test in fbcode since we don't have TARGET file for kernels",
+)
 class TestMarlinQQQ(TestCase):
     def setUp(self):
         super().setUp()