Update subdataset.py

janelia-cellmap · Nov 4, 2024 · 6371af1 · 6371af1
1 parent 8bb0d97
commit 6371af1
Showing 1 changed file with 1 addition and 5 deletions.
diff --git a/src/cellmap_data/subdataset.py b/src/cellmap_data/subdataset.py
@@ -10,11 +10,7 @@ class CellMapSubset(Subset):
     This subclasses PyTorch Subset to wrap a CellMapDataset or CellMapMultiDataset object under a common API, which can be used for dataloading. It maintains the same API as the Subset class. It retrieves raw and groundtruth data from a CellMapDataset or CellMapMultiDataset object.
     """
 
-    def __init__(
-        self,
-        dataset: CellMapDataset | CellMapMultiDataset,
-        indices: Sequence[int],
-    ) -> None:
+    def __init__(self, dataset: CellMapDataset | CellMapMultiDataset, indices: Sequence[int], ) -> None:
         """
         Args:
             dataset: CellMapDataset | CellMapMultiDataset