Fix ruff after #2724

Author: andrewor14

test/quantization/pt2e/test_duplicate_dq.py

@@ -11,7 +11,6 @@
 from typing import Any
 
 import torch
-from torch.export import export_for_training
 from torch.testing._internal.common_quantization import QuantizationTestCase
 from torch.testing._internal.common_utils import IS_WINDOWS, run_tests
 

test/quantization/pt2e/test_quantize_pt2e.py

@@ -19,7 +19,6 @@
     per_channel_weight_observer_range_neg_127_to_127,
     weight_observer_range_neg_127_to_127,
 )
-from torch.export import export_for_training
 from torch.fx import Node
 from torch.testing._internal.common_quantization import (
     NodeSpec as ns,

test/quantization/pt2e/test_quantize_pt2e_qat.py

@@ -18,7 +18,6 @@
     default_symmetric_qnnpack_qat_qconfig,
 )
 from torch.ao.quantization.quantize_fx import prepare_qat_fx
-from torch.export import export_for_training
 from torch.testing._internal.common_cuda import TEST_CUDA
 from torch.testing._internal.common_quantization import (
     NodeSpec as ns,

test/quantization/pt2e/test_representation.py

@@ -11,7 +11,6 @@
 
 import torch
 from torch._higher_order_ops.out_dtype import out_dtype  # noqa: F401
-from torch.export import export_for_training
 from torch.testing._internal.common_quantization import (
     NodeSpec as ns,
 )

test/quantization/pt2e/test_x86inductor_quantizer.py

@@ -12,7 +12,6 @@
 
 import torch
 import torch.nn as nn
-from torch.export import export_for_training
 from torch.testing._internal.common_quantization import (
     NodeSpec as ns,
 )

test/quantization/test_qat.py

@@ -1805,9 +1805,6 @@ def test_qat_api_deprecation(self):
                     str(w.message),
                 )
 
-    @unittest.skipIf(
-        not TORCH_VERSION_AT_LEAST_2_4, "skipping when torch version is 2.4 or lower"
-    )
     def test_qat_api_convert_no_quantization(self):
         """
         Test that `QATConfig(step="convert")` swaps back to nn modules without quantization.

torchao/testing/pt2e/utils.py

@@ -15,7 +15,6 @@
     _convert_to_reference_decomposed_fx,
     prepare_fx,
 )
-from torch.export import export_for_training
 from torch.testing._internal.common_quantization import (
     NodeSpec,
     QuantizationTestCase,