add choices [csv,xml,html] to argparse

rapidsai · Dec 14, 2021 · 1d1b8a0 · 1d1b8a0
1 parent 95324e9
commit 1d1b8a0
Showing 1 changed file with 5 additions and 1 deletion.
diff --git a/cpp/scripts/sort_ninja_log.py b/cpp/scripts/sort_ninja_log.py
@@ -12,7 +12,11 @@
     "log_file", type=str, default=".ninja_log", help=".ninja_log file"
 )
 parser.add_argument(
-    "--fmt", type=str, default="csv", help="output format (csv, xml, html)"
+    "--fmt",
+    type=str,
+    default="csv",
+    choices=["csv", "xml", "html"],
+    help="output format (to stdout)",
 )
 parser.add_argument(
     "--msg",