Commit 30cedf0c authored by baberabb's avatar baberabb
Browse files

type annotations 3.8 compatible

parent f4c1b9d6
...@@ -4,7 +4,7 @@ import sympy ...@@ -4,7 +4,7 @@ import sympy
from sympy.parsing.latex import parse_latex from sympy.parsing.latex import parse_latex
import signal import signal
from lm_eval.logger import eval_logger from lm_eval.logger import eval_logger
from typing import Optional from typing import Optional, List, Dict
# taken from # taken from
...@@ -69,7 +69,7 @@ def process_docs(dataset: datasets.Dataset) -> datasets.Dataset: ...@@ -69,7 +69,7 @@ def process_docs(dataset: datasets.Dataset) -> datasets.Dataset:
return dataset.map(_process_doc) return dataset.map(_process_doc)
def process_results(doc: dict, results: list[str]) -> dict[str, int]: def process_results(doc: dict, results: List[str]) -> Dict[str, int]:
candidates = results[0] candidates = results[0]
unnormalized_answer = get_unnormalized_answer(candidates) unnormalized_answer = get_unnormalized_answer(candidates)
......
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