Skip to content

Commit 09ae578

Browse files
Apply automatic Ruff fixes
1 parent 3a96640 commit 09ae578

File tree

1 file changed

+4
-4
lines changed

1 file changed

+4
-4
lines changed

test/quantization/test_quant_api.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -15,25 +15,25 @@
1515
import torch
1616
from torch.ao.quantization.quantize_pt2e import convert_pt2e, prepare_pt2e
1717
from torch.ao.quantization.quantizer.xnnpack_quantizer import (
18-
get_symmetric_quantization_config,
1918
XNNPACKQuantizer,
19+
get_symmetric_quantization_config,
2020
)
2121
from torch.testing._internal import common_utils
2222
from torch.testing._internal.common_utils import TestCase
2323

2424
from torchao import quantize_
25-
from torchao._models.llama.model import prepare_inputs_for_model, Transformer
25+
from torchao._models.llama.model import Transformer, prepare_inputs_for_model
2626
from torchao._models.llama.tokenizer import get_tokenizer
2727
from torchao.dtypes import AffineQuantizedTensor
2828
from torchao.quantization import LinearActivationQuantizedTensor
2929
from torchao.quantization.quant_api import (
30+
Quantizer,
31+
TwoStepQuantizer,
3032
_replace_with_custom_fn_if_matches_filter,
3133
int4_weight_only,
3234
int8_dynamic_activation_int4_weight,
3335
int8_dynamic_activation_int8_weight,
3436
int8_weight_only,
35-
Quantizer,
36-
TwoStepQuantizer,
3737
)
3838
from torchao.quantization.quant_primitives import MappingType
3939
from torchao.quantization.subclass import (

0 commit comments

Comments
 (0)