nits

38e3b73f · baberabb · 8b74beaa · 38e3b73f
Commit 38e3b73f authored Dec 06, 2023 by baberabb
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

scripts/model_comparator.py scripts/model_comparator.py +1 -1

No files found.
--- a/scripts/model_comparator.py
+++ b/scripts/model_comparator.py
@@ -61,7 +61,7 @@ def parse_args():
    parser.add_argument("--tasks", type=str, default="arc_easy,hellaswag")
    parser.add_argument(
        "--limit",
-        type=int,
+        type=float,
        default=100,
    )
    parser.add_argument(