nit

31631407 · Baber · 29ac037d · 31631407
Commit 31631407 authored Dec 10, 2024 by Baber
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

lm_eval/tasks/llama3/base/utils.py lm_eval/tasks/llama3/base/utils.py +1 -1

No files found.
--- a/lm_eval/tasks/llama3/base/utils.py
+++ b/lm_eval/tasks/llama3/base/utils.py
 import datasets


-def process_docs(dataset: datasets.Dataset) -> datasets.Dataset:
+def process_arc_c_docs(dataset: datasets.Dataset) -> datasets.Dataset:
    COLUMNS = dataset.column_names

    def map_(doc):