Add `description_dict` to results config

d34ae3cf · Jonathan Tow · 70f92733 · d34ae3cf
Commit d34ae3cf authored Dec 21, 2021 by Jonathan Tow
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 1 deletion

lm_eval/evaluator.py lm_eval/evaluator.py +2 -1

No files found.
--- a/lm_eval/evaluator.py
+++ b/lm_eval/evaluator.py
@@ -62,7 +62,8 @@ def simple_evaluate(model, model_args, task_names,
        "device": device,
        "no_cache": no_cache,
        "limit": limit,
-        "bootstrap_iters": bootstrap_iters
+        "bootstrap_iters": bootstrap_iters,
+        "description_dict": description_dict
    }

    return results