Skip to content

Commit

Permalink
The arg_str should be placed after input_model and output_model
Browse files Browse the repository at this point in the history
  • Loading branch information
whyb committed Nov 27, 2024
1 parent 69b872f commit b7f4605
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tests/test_onnxslim.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
Expand Down

0 comments on commit b7f4605

Please sign in to comment.