Skip to content

Commit f27fbbd

Browse files
committed
Database: add alias property
1 parent cda34a4 commit f27fbbd

File tree

1 file changed

+7
-3
lines changed

1 file changed

+7
-3
lines changed

lazy_dataset/database.py

Lines changed: 7 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -81,6 +81,10 @@ def data(self):
8181
raise NotImplementedError(
8282
f'Override this property in {self.__class__.__name__}!')
8383

84+
@property
85+
def alias(self):
86+
return self.data.setdefault('alias', {})
87+
8488
@property
8589
def dataset_names(self):
8690
"""
@@ -90,7 +94,7 @@ def dataset_names(self):
9094
return tuple(
9195
self.data['datasets'].keys()
9296
) + tuple(
93-
self.data.get('alias', {}).keys()
97+
self.alias.keys()
9498
)
9599

96100
def get_examples(self, dataset_name):
@@ -107,8 +111,8 @@ def get_examples(self, dataset_name):
107111
108112
"""
109113
try:
110-
if dataset_name in self.data.get('alias', []):
111-
dataset_names = self.data['alias'][dataset_name]
114+
if dataset_name in self.alias:
115+
dataset_names = self.alias[dataset_name]
112116
examples = {}
113117
for name in dataset_names:
114118
examples_new = self.data['datasets'][name]

0 commit comments

Comments
 (0)