Black

NKI-AI · Apr 3, 2024 · 1b5b1d5 · 1b5b1d5
1 parent 1ea6243
commit 1b5b1d5
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/ahcore/data/dataset.py b/ahcore/data/dataset.py
@@ -86,10 +86,12 @@ def __len__(self) -> int:
         return self.cumulative_sizes[-1]
 
     @overload
-    def __getitem__(self, index: int) -> DlupDatasetSample: ...
+    def __getitem__(self, index: int) -> DlupDatasetSample:
+        ...
 
     @overload
-    def __getitem__(self, index: slice) -> list[DlupDatasetSample]: ...
+    def __getitem__(self, index: slice) -> list[DlupDatasetSample]:
+        ...
 
     def __getitem__(self, index: Union[int, slice]) -> DlupDatasetSample | list[DlupDatasetSample]:
         """Returns the sample at the given index."""