update

pytorch · Nov 29, 2024 · 3ea0f9b · 3ea0f9b
1 parent 5194c8c
commit 3ea0f9b
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/userbenchmark/test_bench/install.py b/userbenchmark/test_bench/install.py
@@ -52,7 +52,7 @@ def install_test_bench_requirements():
 
     model_paths = _filter_model_paths(args.models, args.skip, args.canary)
     for path in model_paths:
-        print(f"Installing {os.path.basename(path)}...")
+        print(f"Installing {os.path.basename(path)}...", end="", flush=True)
         install_py_path = os.path.join(path, "install.py")
         requirements_txt_path = os.path.join(path, "requirements.txt")
 
@@ -64,13 +64,13 @@ def install_test_bench_requirements():
         elif os.path.exists(requirements_txt_path):
             success, error_msg = pip_install_requirements(path, continue_on_fail)
         else:
-            print(f"SKipped: {os.path.basename(path)}")
+            print("SKipped")
             continue
 
         if success:
             print("OK")
         else:
-            print("Failed to install test_bench requirements:")
+            print("Failed")
             print(error_msg)
             if not continue_on_fail:
                 sys.exit(-1)