Skip to content

Commit de601b8

Browse files
committed
Fix: reverted 1 line from rebase + quantize.quantize -> quantize
Signed-off-by: gcunhase <[email protected]>
1 parent 7cbfb56 commit de601b8

File tree

2 files changed

+1
-2
lines changed

2 files changed

+1
-2
lines changed

tests/_test_utils/onnx/quantization/lib_test_models.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -675,7 +675,6 @@ def build_conv_batchnorm_sig_mul_model():
675675
return model_inferred
676676

677677

678-
def build_conv_act_pool_model():
679678
def build_conv_act_pool_model(include_reshape_node=False):
680679
# Define your model inputs and outputs
681680
input_names = ["input_0"]

tests/unit/onnx/test_qdq_rules_int8.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -159,7 +159,7 @@ def test_conv_act_pool_int8(tmp_path, include_reshape_node):
159159
onnx_path = os.path.join(tmp_path, f"conv_act_pool_model_{include_reshape_node}.onnx")
160160
save_onnx(onnx_model, onnx_path)
161161

162-
quantize.quantize(onnx_path, quantize_mode="int8", high_precision_dtype="fp16")
162+
quantize(onnx_path, quantize_mode="int8", high_precision_dtype="fp16")
163163

164164
# Output model should be produced in the same tmp_path
165165
output_onnx_path = onnx_path.replace(".onnx", ".quant.onnx")

0 commit comments

Comments
 (0)