Addresses #463

maps-as-data · Jul 17, 2024 · f8265ee · f8265ee
1 parent 8bf049a
commit f8265ee
Showing 1 changed file with 8 additions and 4 deletions.
diff --git a/mapreader/classify/load_annotations.py b/mapreader/classify/load_annotations.py
@@ -18,10 +18,12 @@
 
 
 class AnnotationsLoader:
+    """
+    A class for loading annotations and preparing datasets and dataloaders for
+    use in training/validation of a model.
+    """
+
     def __init__(self):
-        """
-        A Class for loading annotations and preparing datasets and dataloaders for use in training/validation of a model.
-        """
         self.annotations = pd.DataFrame()
         self.reviewed = pd.DataFrame()
         self.patch_paths_col = None
@@ -41,7 +43,9 @@ def load(
         scramble_frame: bool | None = False,
         reset_index: bool | None = False,
     ):
-        """Loads annotations from a csv file or dataframe and can be used to set the ``patch_paths_col`` and ``label_col`` attributes.
+        """
+        Loads annotations from a csv file or dataframe and can be used to set
+        the ``patch_paths_col`` and ``label_col`` attributes.
 
         Parameters
         ----------