Skip to content

Commit 2feab84

Browse files
committed
[Test] override_batchsize -> override_inpsize for cleanup
1 parent 038f21e commit 2feab84

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

tests/transformation/test_introduce_quantnode.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -81,7 +81,7 @@ def download_model(test_model, do_cleanup=False, return_modelwrapper=False):
8181
urllib.request.urlretrieve(qonnx_url, dl_file)
8282
if do_cleanup:
8383
out_file = dl_dir + f"/{test_model}_clean.onnx"
84-
cleanup(dl_file, out_file=out_file, override_batchsize=1)
84+
cleanup(dl_file, out_file=out_file, override_inpsize=1)
8585
ret = out_file
8686
if return_modelwrapper:
8787
ret = ModelWrapper(ret)

0 commit comments

Comments
 (0)