From 4939ecb123b0ffc880de9dc3f3b81a0c67aac4c2 Mon Sep 17 00:00:00 2001 From: rhoadesScholar Date: Fri, 10 Jan 2025 16:10:18 -0500 Subject: [PATCH] fix: Change classes parameter type from Iterable to Sequence in CellMapDataLoader --- src/cellmap_data/dataloader.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/cellmap_data/dataloader.py b/src/cellmap_data/dataloader.py index 894668e..c77be02 100644 --- 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,