|
@@ -142,7 +142,7 @@ class BaseAnnotations(abc.ABC):
|
|
return self._uuids(self.test_split)
|
|
return self._uuids(self.test_split)
|
|
|
|
|
|
def new_train_test_datasets(self, dataset_cls=Dataset, **kwargs):
|
|
def new_train_test_datasets(self, dataset_cls=Dataset, **kwargs):
|
|
- return (self.new_dataset(subset, dataset_cls) for subset in ["train", "test"])
|
|
|
|
|
|
+ return (self.new_dataset(subset, dataset_cls, **kwargs) for subset in ["train", "test"])
|
|
|
|
|
|
def new_dataset(self, subset=None, dataset_cls=Dataset, **kwargs):
|
|
def new_dataset(self, subset=None, dataset_cls=Dataset, **kwargs):
|
|
if subset is not None:
|
|
if subset is not None:
|