From f8265eee0d058fee6d01bedb96ffb24c43dd7204 Mon Sep 17 00:00:00 2001 From: Kalle Westerling Date: Wed, 17 Jul 2024 15:26:33 +0100 Subject: [PATCH] Addresses #463 --- mapreader/classify/load_annotations.py | 12 ++++++++---- 1 file changed, 8 insertions(+), 4 deletions(-) diff --git a/mapreader/classify/load_annotations.py b/mapreader/classify/load_annotations.py index 6292b211..123c9b35 100644 --- 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 ----------