add default option

mosaicml · Feb 12, 2024 · 124b60a · 124b60a
1 parent 0ffab21
commit 124b60a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/llmfoundry/utils/builders.py b/llmfoundry/utils/builders.py
@@ -52,8 +52,8 @@ def build_evaluators(
     tokenizer: PreTrainedTokenizerBase,
     device_eval_batch_size: int,
     icl_seq_len: int,
-    fewshot_random_seed: Optional[int],
     icl_subset_num_batches: Optional[int],
+    fewshot_random_seed: Optional[int] = None,
 ) -> Tuple[List[Evaluator], List[str], Optional[EvalGauntlet]]:
 
     evaluators = []