set type to int for ctx lens

jbloomAus · Oct 18, 2024 · e4b5be6 · e4b5be6
1 parent d2bebbc
commit e4b5be6
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/sae_lens/evals.py b/sae_lens/evals.py
@@ -883,6 +883,7 @@ def process_results(eval_results: list[defaultdict[Any, Any]], output_dir: str):
     arg_parser.add_argument(
         "--ctx_lens",
         nargs="+",
+        type=int,
         default=[128],
         help="Context lengths to evaluate on.",
     )