fix: Change classes parameter type from Iterable to Sequence in CellM…

…apDataLoader
janelia-cellmap · Jan 10, 2025 · 4939ecb · 4939ecb
1 parent 7530679
commit 4939ecb
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/cellmap_data/dataloader.py b/src/cellmap_data/dataloader.py
@@ -3,7 +3,7 @@
 from .dataset import CellMapDataset
 from .multidataset import CellMapMultiDataset
 from .dataset_writer import CellMapDatasetWriter
-from typing import Callable, Iterable, Optional, Sequence
+from typing import Callable, Optional, Sequence
 
 
 class CellMapDataLoader:
@@ -31,7 +31,7 @@ class CellMapDataLoader:
     def __init__(
         self,
         dataset: CellMapMultiDataset | CellMapDataset | Subset | CellMapDatasetWriter,
-        classes: Iterable[str],
+        classes: Sequence[str],
         batch_size: int = 1,
         num_workers: int = 0,
         weighted_sampler: bool = False,