Replace underscores with dashes in argument names

togethercomputer · Apr 24, 2024 · 0d7300e · 0d7300e
1 parent 44d9ae7
commit 0d7300e
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/examples/tokenize_data.py b/examples/tokenize_data.py
@@ -95,7 +95,7 @@ def process_data(args: argparse.Namespace) -> None:
         "--max-seq-length", type=int, default=2048, help="Maximum sequence length"
     )
     parser.add_argument(
-        "--add_labels",
+        "--add-labels",
         action="store_true",
         help="Whether to add loss labels from padding tokens",
     )
@@ -106,7 +106,7 @@ def process_data(args: argparse.Namespace) -> None:
         help="Tokenizer name (for example, togethercomputer/LLaMA-2-7B-32K)",
     )
     parser.add_argument(
-        "--out_filename",
+        "--out-filename",
         default="processed_dataset.parquet",
         help="Name of the Parquet file to save (should have .parquet extension)",
     )