Unverified Commit d289277d authored by Lintang Sutawika's avatar Lintang Sutawika Committed by GitHub
Browse files

Update __init__.py

Should fix a potential error if a dataset has no `dataset_name` argument.

cc: @haileyschoelkopf
parent 070b6b9c
......@@ -14,7 +14,10 @@ from lm_eval.api.registry import (
def get_task_name_from_config(task_config):
return "{dataset_path}_{dataset_name}".format(**task_config)
if "dataset_name" not in task_config:
return "{dataset_path}_{dataset_name}".format(**task_config)
else:
return "{dataset_path}".format(**task_config)
def include_task_folder(task_dir):
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment