From 312c5e5532f0e0470bf47f77a6243e02a61da530 Mon Sep 17 00:00:00 2001 From: Yichen Xu Date: Wed, 1 Dec 2021 03:14:24 +0800 Subject: [PATCH] Fix type signature of read_problems (#9) --- human_eval/data.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/human_eval/data.py b/human_eval/data.py index 4fd8905..ed49794 100644 --- a/human_eval/data.py +++ b/human_eval/data.py @@ -8,7 +8,7 @@ HUMAN_EVAL = os.path.join(ROOT, "..", "data", "HumanEval.jsonl.gz") -def read_problems(evalset_file: str = HUMAN_EVAL) -> Iterable[Dict[str, Dict]]: +def read_problems(evalset_file: str = HUMAN_EVAL) -> Dict[str, Dict]: return {task["task_id"]: task for task in stream_jsonl(evalset_file)}