fix the context manager call

axolotl-ai-cloud · Sep 4, 2023 · 83d904a · 83d904a
1 parent 5e4a760
commit 83d904a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/axolotl/utils/callbacks.py b/src/axolotl/utils/callbacks.py
@@ -210,7 +210,7 @@ def tokenize_evals(example):
             "subject": example["subject"],
         }
 
-    with dist_state.main_process_first:
+    with dist_state.main_process_first():
         bench_dataset = bench_dataset.map(tokenize_evals)
         bench_dataset = bench_dataset.filter(lambda x: x["labels"][-2] in abcd_idx)