Skip to content

GitLab

Unverified Commit 1bf97c9e authored by Leo Gao's avatar Leo Gao Committed by GitHub
Browse files

Make *_docs not abstract

parent e48b6082
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 6 deletions
lm_eval/base.py
View file @ 1bf97c9e
...@@ -82,22 +82,19 @@ class Dataset(abc.ABC): ...@@ -82,22 +82,19 @@ class Dataset(abc.ABC):
"""Whether the task has a test set""" """Whether the task has a test set"""
pass pass
@abc.abstractmethod
def training_docs(self): def training_docs(self):
""" """
:return: Iterable[obj] :return: Iterable[obj]
A iterable of any object, that doc_to_text can handle A iterable of any object, that doc_to_text can handle
""" """
pass return []
@abc.abstractmethod
def validation_docs(self): def validation_docs(self):
pass return []
@abc.abstractmethod
def test_docs(self): def test_docs(self):
pass return []
def fewshot_examples(self, k): def fewshot_examples(self, k):
if self._traindocs is None: if self._traindocs is None:
......
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