From 6ef34761d1b8d5fb94565105ed0f8eae7390fa71 Mon Sep 17 00:00:00 2001 From: Fabian Isensee Date: Wed, 27 Mar 2024 10:08:57 +0100 Subject: [PATCH] remove dict assertion --- nnunetv2/training/dataloading/base_data_loader.py | 1 - 1 file changed, 1 deletion(-) diff --git a/nnunetv2/training/dataloading/base_data_loader.py b/nnunetv2/training/dataloading/base_data_loader.py index 6a6a49f1f..21bddecf8 100644 --- a/nnunetv2/training/dataloading/base_data_loader.py +++ b/nnunetv2/training/dataloading/base_data_loader.py @@ -19,7 +19,6 @@ def __init__(self, pad_sides: Union[List[int], Tuple[int, ...], np.ndarray] = None, probabilistic_oversampling: bool = False): super().__init__(data, batch_size, 1, None, True, False, True, sampling_probabilities) - assert isinstance(data, nnUNetDataset), 'nnUNetDataLoaderBase only supports dictionaries as data' self.indices = list(data.keys()) self.oversample_foreground_percent = oversample_foreground_percent