Fix type signature of read_problems (#9)

openai · Nov 30, 2021 · 312c5e5 · 312c5e5
1 parent fa06031
commit 312c5e5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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)}