re-enable MultiChoice on --tasks

b5ce475d · haileyschoelkopf · 2e2f28a5 · b5ce475d
Commit b5ce475d authored Aug 17, 2023 by haileyschoelkopf
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

main.py main.py +1 -1

No files found.
--- a/main.py
+++ b/main.py
@@ -24,7 +24,7 @@ def parse_args():
        help="String arguments for model, e.g. `pretrained=EleutherAI/pythia-160m,dtype=float32`",
    )
    parser.add_argument(
-        "--tasks", default=None  # , choices=utils.MultiChoice(sorted(ALL_TASKS))
+        "--tasks", default=None, choices=utils.MultiChoice(sorted(ALL_TASKS))
    )
    parser.add_argument(
        "--num_fewshot",