Skip to content

Commit 4564eb9

Browse files
committed
WIP: usage example
1 parent e5cef8d commit 4564eb9

File tree

2 files changed

+4
-1
lines changed

2 files changed

+4
-1
lines changed

datasets/squad/squad.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -20,6 +20,7 @@
2020
import json
2121

2222
import datasets
23+
from datasets.tasks import QuestionAnswering
2324

2425

2526
logger = datasets.logging.get_logger(__name__)
@@ -98,6 +99,9 @@ def _info(self):
9899
supervised_keys=None,
99100
homepage="https://rajpurkar.github.io/SQuAD-explorer/",
100101
citation=_CITATION,
102+
task_templates=[
103+
QuestionAnswering(),
104+
],
101105
)
102106

103107
def _split_generators(self, dl_manager):

src/datasets/tasks/base.py

Lines changed: 0 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -5,7 +5,6 @@
55
from ..features import Features
66

77

8-
@dataclass
98
class TaskTemplate(abc.ABC):
109
task: str
1110
input_schema: Features

0 commit comments

Comments
 (0)