updated docstring

kaiko-ai · Dec 6, 2024 · c1adf75 · c1adf75
1 parent daab418
commit c1adf75
Showing 1 changed file with 3 additions and 7 deletions.
diff --git a/src/eva/core/data/datasets/base.py b/src/eva/core/data/datasets/base.py
@@ -63,18 +63,14 @@ def __getitem__(self, index: int):
         """Retrieves the item at the given index.
 
         Args:
-            index (int): Index
+            index: Index of the item to retrieve.
 
         Returns:
-            Any: The data at the given index
+            The data at the given index.
         """
         raise NotImplementedError
 
     @abc.abstractmethod
     def __len__(self) -> int:
-        """Returns the length of the dataset.
-
-        Returns:
-            int: Length of the dataset
-        """
+        """Returns the length of the dataset."""
         raise NotImplementedError