From b7f4605c3b44bf78eea93c5410d21716956f4f18 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?=E5=BC=B5=E5=B0=8F=E5=87=A1?= <2672931+whyb@users.noreply.github.com> Date: Wed, 27 Nov 2024 10:39:07 +0800 Subject: [PATCH] The arg_str should be placed after input_model and output_model --- tests/test_onnxslim.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_onnxslim.py b/tests/test_onnxslim.py index 5536376..7f29c35 100644 --- a/tests/test_onnxslim.py +++ b/tests/test_onnxslim.py @@ -19,7 +19,7 @@ def __test_command_basic(self, request, in_model_path=FILENAME, out_model_name=" out_model_path = os.path.join(tempdir, out_model_name) slim(in_model_path, out_model_path) slim(in_model_path, out_model_path, model_check=True) - command = f'onnxslim {arg_str} "{in_model_path}" "{out_model_path}"' + command = f'onnxslim "{in_model_path}" "{out_model_path}" {arg_str}' result = subprocess.run(command, shell=True, capture_output=True, text=True) output = result.stderr.strip() # Assert the expected return code