Vittal-M commited on
Commit
212eb34
·
verified ·
1 Parent(s): 9fe5ab4

Update scripts/run_pipeline.py

Browse files
Files changed (1) hide show
  1. scripts/run_pipeline.py +2 -1
scripts/run_pipeline.py CHANGED
@@ -62,10 +62,11 @@ def main() -> None:
62
  parser.add_argument("--no-eval", action="store_true", help="Skip benchmark evaluation")
63
  parser.add_argument("--workers", type=int, default=4, help="Number of parallel workers")
64
  parser.add_argument("--scenarios", type=int, default=None, help="Override number of scenarios")
 
65
  args = parser.parse_args()
66
 
67
  n_scenarios = args.scenarios or (50 if args.quick else 1000)
68
- n_eval_seeds = 20 if args.quick else 1000
69
  n_workers = args.workers
70
 
71
  t_start = time.time()
 
62
  parser.add_argument("--no-eval", action="store_true", help="Skip benchmark evaluation")
63
  parser.add_argument("--workers", type=int, default=4, help="Number of parallel workers")
64
  parser.add_argument("--scenarios", type=int, default=None, help="Override number of scenarios")
65
+ parser.add_argument("--eval-seeds", type=int, default=None, help="Override number of evaluation seeds")
66
  args = parser.parse_args()
67
 
68
  n_scenarios = args.scenarios or (50 if args.quick else 1000)
69
+ n_eval_seeds = args.eval_seeds or (20 if args.quick else 1000)
70
  n_workers = args.workers
71
 
72
  t_start = time.time()