Commit 43c91b5c authored by haileyschoelkopf's avatar haileyschoelkopf
Browse files

hotfix bugs

parent 265a7cde
...@@ -631,7 +631,7 @@ class ConfigurableTask(Task): ...@@ -631,7 +631,7 @@ class ConfigurableTask(Task):
if self.fewshot_docs() is not None: if self.fewshot_docs() is not None:
self.sampler = samplers.get_sampler( self.sampler = samplers.get_sampler(
self.config.fewshot_config.get("sampler", "default") self.config.fewshot_config.get("sampler", "default") if self.config.fewshot_config else "default"
)(list(self.fewshot_docs()), self, rnd=random.Random(1234)) )(list(self.fewshot_docs()), self, rnd=random.Random(1234))
if self.has_test_docs(): if self.has_test_docs():
......
import datasets import datasets
import re import re
import sympy
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, List, Dict from typing import Optional, List, Dict
try:
import sympy
from sympy.parsing.latex import parse_latex
except ModuleNotFoundError:
raise Exception(
"`sympy` is required for generating translation task prompt templates. \
please install sympy via pip install lm-eval[math] or pip install -e .[math]",
)
# taken from # taken from
# https://github.com/wellecks/lm-evaluation-harness/blob/master/lm_eval/tasks/minerva_math.py # https://github.com/wellecks/lm-evaluation-harness/blob/master/lm_eval/tasks/minerva_math.py
......
...@@ -45,6 +45,14 @@ def escaped_split(text, sep_char, maxsplit=-1): ...@@ -45,6 +45,14 @@ def escaped_split(text, sep_char, maxsplit=-1):
return re.split(r"(?<!\\)" + sep_char, text, maxsplit) return re.split(r"(?<!\\)" + sep_char, text, maxsplit)
def handle_arg_string(arg):
if arg.lower() == "true":
return True
elif arg.lower() == "false":
return False
return arg
def simple_parse_args_string(args_string): def simple_parse_args_string(args_string):
""" """
Parses something like Parses something like
...@@ -55,7 +63,7 @@ def simple_parse_args_string(args_string): ...@@ -55,7 +63,7 @@ def simple_parse_args_string(args_string):
if not args_string: if not args_string:
return {} return {}
arg_list = [arg for arg in args_string.split(",") if arg] arg_list = [arg for arg in args_string.split(",") if arg]
args_dict = {k: v for k, v in [arg.split("=") for arg in arg_list]} args_dict = {k: handle_arg_string(v) for k, v in [arg.split("=") for arg in arg_list]}
return args_dict return args_dict
......
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