Skip to content

Commit a8c88f4

Browse files
committed
fix: use configurations
1 parent b98d938 commit a8c88f4

File tree

3 files changed

+8
-7
lines changed

3 files changed

+8
-7
lines changed

.gitignore

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -1,4 +1,5 @@
11
*.egg-info
2+
.DS_Store
23
.vscode
34
__pycache__
45
site

src/datamaestro_text/config/com/github/prdwb/orconvqa.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -49,9 +49,9 @@ def preprocessed(train, dev, test):
4949
Each dataset is an instance of :class:`datamaestro_text.data.conversation.OrConvQADataset`
5050
"""
5151
return {
52-
"train": OrConvQADataset(path=train),
53-
"validation": OrConvQADataset(path=dev),
54-
"test": OrConvQADataset(path=test),
52+
"train": OrConvQADataset.C(path=train),
53+
"validation": OrConvQADataset.C(path=dev),
54+
"test": OrConvQADataset.C(path=test),
5555
}
5656

5757

src/datamaestro_text/datasets/irds/datasets.py

Lines changed: 4 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -65,7 +65,7 @@ def download(self, force=False):
6565
return True
6666

6767
def _prepare(self, download=False) -> Documents:
68-
return AdhocAssessments(id=self.fullid)
68+
return AdhocAssessments.C(id=self.fullid)
6969

7070

7171
class QueriesDataset(Dataset):
@@ -78,7 +78,7 @@ def download(self, force=False):
7878
return True
7979

8080
def _prepare(self, download=False) -> Documents:
81-
return Topics(id=self.fullid)
81+
return Topics.C(id=self.fullid)
8282

8383

8484
# class ScoredDocuments(Dataset):
@@ -96,7 +96,7 @@ def download(self, force=False):
9696
return True
9797

9898
def _prepare(self, download=False) -> Documents:
99-
return Documents(id=self.fullid)
99+
return Documents.C(id=self.fullid)
100100

101101

102102
class TrainingTripletsDataset(Dataset):
@@ -123,7 +123,7 @@ class Collection(Dataset):
123123
topics: QueriesDataset
124124

125125
def _prepare(self, download=False) -> Documents:
126-
return Adhoc(
126+
return Adhoc.C(
127127
id=self.fullid,
128128
topics=self.topics.prepare(download),
129129
assessments=self.assessments.prepare(download),

0 commit comments

Comments
 (0)