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

type annotations 3.8 compatible

parent f4c1b9d6
......@@ -4,7 +4,7 @@ import sympy
from sympy.parsing.latex import parse_latex
import signal
from lm_eval.logger import eval_logger
from typing import Optional
from typing import Optional, List, Dict
# taken from
......@@ -69,7 +69,7 @@ def process_docs(dataset: datasets.Dataset) -> datasets.Dataset:
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]
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