Pārlūkot izejas kodu

important fix for dataset initialization

Dimitri Korsch 2 gadi atpakaļ
vecāks
revīzija
724a0daba1
1 mainītis faili ar 1 papildinājumiem un 1 dzēšanām
  1. 1 1
      cvdatasets/annotation/base.py

+ 1 - 1
cvdatasets/annotation/base.py

@@ -142,7 +142,7 @@ class BaseAnnotations(abc.ABC):
 		return self._uuids(self.test_split)
 
 	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):
 		if subset is not None: