Bugfix: pass on sampler to ImageCollectionDataset in convenience func…

…tion
constantinpape · Jul 15, 2023 · 48a2936 · 48a2936
1 parent 6fcb386
commit 48a2936
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/torch_em/segmentation.py b/torch_em/segmentation.py
@@ -291,6 +291,7 @@ def default_segmentation_dataset(
             label_transform2=label_transform2,
             transform=transform,
             n_samples=n_samples,
+            sampler=sampler,
             dtype=dtype,
             label_dtype=label_dtype,
         )