Fix type signature of read_problems (#9)

312c5e55 · Yichen Xu · GitHub · fa06031e · 312c5e55
Unverified Commit 312c5e55 authored Dec 01, 2021 by Yichen Xu Committed by GitHub Nov 30, 2021
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

human_eval/data.py human_eval/data.py +1 -1

No files found.
--- a/human_eval/data.py
+++ b/human_eval/data.py
@@ -8,7 +8,7 @@ ROOT = os.path.dirname(os.path.abspath(__file__))
 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)}