소스 검색

important fix for dataset initialization

Dimitri Korsch 2 년 전
부모
커밋
724a0daba1
1개의 변경된 파일1개의 추가작업 그리고 1개의 파일을 삭제
  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: