Commit b3a8c985 authored by lintangsutawika's avatar lintangsutawika
Browse files

reformat

parent 881fcf15
...@@ -114,7 +114,12 @@ def simple_evaluate( ...@@ -114,7 +114,12 @@ def simple_evaluate(
task_dict = lm_eval.tasks.get_task_dict(tasks) task_dict = lm_eval.tasks.get_task_dict(tasks)
for task_name in task_dict.keys(): for task_name in task_dict.keys():
config = task_dict[task_name]._config
task_obj = task_dict[task_name]
if type(task_obj) == tuple:
group, task_obj = task_obj
config = task_obj._config
if num_fewshot is not None: if num_fewshot is not None:
if config["num_fewshot"] > 0: if config["num_fewshot"] > 0:
default_num_fewshot = config["num_fewshot"] default_num_fewshot = config["num_fewshot"]
......
...@@ -6,7 +6,7 @@ from . import dummy ...@@ -6,7 +6,7 @@ from . import dummy
try: try:
import anthropic import anthropic
from . import anthropic_llms from . import anthropic_llms
except Exception as err: except Exception:
raise "anthropic library is not yet installed" raise "anthropic library is not yet installed"
# TODO: implement __all__ # TODO: implement __all__
...@@ -28,7 +28,7 @@ setuptools.setup( ...@@ -28,7 +28,7 @@ setuptools.setup(
"License :: OSI Approved :: MIT License", "License :: OSI Approved :: MIT License",
"Operating System :: OS Independent", "Operating System :: OS Independent",
], ],
python_requires=">=3.9", python_requires=">=3.8",
install_requires=[ install_requires=[
"accelerate>=0.18.0", "accelerate>=0.18.0",
"evaluate", "evaluate",
......
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