remove dict assertion

MIC-DKFZ · Mar 27, 2024 · 6ef3476 · 6ef3476
1 parent 0db9560
commit 6ef3476
Showing 1 changed file with 0 additions and 1 deletion.
diff --git 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