Commit 6f98c780 authored by lintangsutawika's avatar lintangsutawika
Browse files

adjusted to follow pre-commit conventions

parent 9329fc9b
......@@ -22,10 +22,12 @@ def get_prompt(prompt_id: str, dataset_name=None, subset_name=None):
eval_logger.info(f"Loading prompt from {category_name} for {dataset_full_name}")
if category_name == "promptsource":
try:
from promptsource.templates import DatasetTemplates
from promptsource.templates import DatasetTemplates
except ModuleNotFoundError:
raise Exception("Tried to load a Promptsource template, but promptsource is not installed ",
"please install promptsource via pip install lm-eval[promptsource] or pip install -e .[promptsource]")
raise Exception(
"Tried to load a Promptsource template, but promptsource is not installed ",
"please install promptsource via pip install lm-eval[promptsource] or pip install -e .[promptsource]",
)
try:
if subset_name is None:
prompts = DatasetTemplates(dataset_name=dataset_name)
......
......@@ -44,6 +44,8 @@ setuptools.setup(
"dev": ["black", "flake8", "pre-commit", "pytest", "pytest-cov"],
"multilingual": ["nagisa>=0.2.7", "jieba>=0.42.1"],
"sentencepiece": ["sentencepiece>=0.1.98", "protobuf>=4.22.1"],
"promptsource": ["promptsource @ git+https://github.com/bigscience-workshop/promptsource.git#egg=promptsource"]
"promptsource": [
"promptsource @ git+https://github.com/bigscience-workshop/promptsource.git#egg=promptsource"
],
},
)
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