update simple online

bespokelabsai · Dec 13, 2024 · f41aa91 · f41aa91
1 parent 585a495
commit f41aa91
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/tests/simple_online.py b/tests/simple_online.py
@@ -19,7 +19,7 @@ def main(args):
         max_requests_per_minute=args.max_requests_per_minute,
         max_tokens_per_minute=args.max_tokens_per_minute,
         max_retries=args.max_retries,
-        require_all_responses=args.require_all_responses,
+        require_all_responses=not args.not_require_all_responses,
     )
 
     dataset = prompter(dataset, batch_cancel=args.cancel)
@@ -45,9 +45,9 @@ def main(args):
     )
     parser.add_argument("--max-retries", type=int, help="Max retries", default=None)
     parser.add_argument(
-        "--require-all-responses",
+        "--not-require-all-responses",
         action="store_true",
-        default=None,
+        default=False,
         help="Require all responses",
     )
     args = parser.parse_args()