This is an automated email from the ASF dual-hosted git repository. andrewzhaoluo pushed a commit to branch ci-docker-staging in repository https://gitbox.apache.org/repos/asf/tvm.git
commit 0bd0c0fc087e312fbe9d7642b7dab89c5e0b814d Author: Andrew Luo <[email protected]> AuthorDate: Thu Jan 13 15:23:32 2022 -0800 formatting --- tests/python/frontend/onnx/test_forward.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/tests/python/frontend/onnx/test_forward.py b/tests/python/frontend/onnx/test_forward.py index d3d868f..eb8c9fd 100644 --- a/tests/python/frontend/onnx/test_forward.py +++ b/tests/python/frontend/onnx/test_forward.py @@ -233,8 +233,7 @@ def verify_with_ort( def quantize_and_verify_with_ort(onnx_model, input_names, input_shapes, target, dev): - from onnxruntime.quantization import (CalibrationDataReader, QuantType, - quantize_static) + from onnxruntime.quantization import CalibrationDataReader, QuantType, quantize_static input_arrays = [np.random.random(shape).astype("float32") for shape in input_shapes]
