Fix the upload file

pytorch · Jun 7, 2024 · 6483b83 · 6483b83
1 parent 9dee985
commit 6483b83
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/userbenchmark/torchao/upload.py b/userbenchmark/torchao/upload.py
@@ -66,6 +66,6 @@ def post_ci_process(output_files: List[str]):
 
 if __name__ == "__main__":
     parser = argparse.ArgumentParser()
-    parser.add_argument("--test-file", type=str, help="Add file to test.")
+    parser.add_argument("--test-files", nargs='+', help="Add files to test.")
     args = parser.parse_args()
-    post_ci_process([args.test_file])
+    post_ci_process(args.test_files)