Add to registry

5ef6ea82 · Niklas Muennighoff · d1b6d12d · 5ef6ea82
Commit 5ef6ea82 authored Feb 12, 2023 by Niklas Muennighoff
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

lm_eval/tasks/__init__.py lm_eval/tasks/__init__.py +1 -0

No files found.
--- a/lm_eval/tasks/__init__.py
+++ b/lm_eval/tasks/__init__.py
@@ -85,6 +85,7 @@ all_translation_benchmarks = {


 TASK_REGISTRY = {
+    "babi": babi.Babi,
    # GLUE
    "cola": glue.CoLA,
    "mnli": glue.MNLI,