Fix formatting and fix a bug in label mapping

Signed-off-by: Albert van Houten <[email protected]>

Author: AlbertvanHouten

library/src/otx/data/dataset/base.py

@@ -158,11 +158,13 @@ def get_idx_list_per_classes(self, use_string_label: bool = False) -> dict[int |
         """Get a dictionary mapping class labels to lists of corresponding sample indices.
 
         Args:
-            use_string_label (bool, optional): If True, use string labels as keys; otherwise, use integer labels. Defaults to False.
+            use_string_label (bool, optional): If True, use string labels as keys; otherwise, use integer labels.
 
         Returns:
-            dict[int | str, list[int]]: A dictionary where each key is a class label (int or str) and each value is a list of sample indices belonging to that class.
+            dict[int | str, list[int]]: A dictionary where each key is a class label (int or str) and each value is a
+            list of sample indices belonging to that class.
         """
+
     @property
     def task_type(self) -> OTXTaskType | None:
         """OTX Task Type for the dataset. Can be None if no task is defined."""

library/src/otx/types/label.py

@@ -360,8 +360,8 @@ def from_dm_label_groups_arrow(cls, dm_label_categories: HierarchicalLabelCatego
         id_to_name_mapping = {item.name: label_names[i] for i, item in enumerate(dm_label_categories.items)}
 
         for i, item in enumerate(dm_label_categories.items):
-            object.__setattr__(dm_label_categories, "name", label_names[i])
-            object.__setattr__(dm_label_categories, "parent", id_to_name_mapping.get(item.parent, item.parent))
+            object.__setattr__(item, "name", label_names[i])
+            object.__setattr__(item, "parent", id_to_name_mapping.get(item.parent, item.parent))
 
         for label_group in dm_label_categories.label_groups:
             object.__setattr__(