From 6371af1033a32144a934a761e9c6dbc16161e6cf Mon Sep 17 00:00:00 2001 From: Jeff Rhoades <37990507+rhoadesScholar@users.noreply.github.com> Date: Mon, 4 Nov 2024 14:10:04 -0500 Subject: [PATCH] Update subdataset.py --- src/cellmap_data/subdataset.py | 6 +----- 1 file changed, 1 insertion(+), 5 deletions(-) diff --git a/src/cellmap_data/subdataset.py b/src/cellmap_data/subdataset.py index 3df9d53..8148596 100644 --- 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