fix pydoctor and typecheck

Author: dtronmans

luxonis_ml/data/datasets/base_dataset.py

@@ -109,6 +109,27 @@ def get_classes(self) -> dict[str, list[str]]:
         """
         ...
 
+    @abstractmethod
+    def set_native_classes(
+        self,
+        native_classes: dict[int, str],
+        source_format: str,
+    ) -> None:
+        """Sets the native class indexing from the original dataset
+        format.
+
+        This stores the mapping from original dataset indices to class
+        names.
+
+        @type native_classes: Dict[int, str]
+        @param native_classes: A dictionary mapping original dataset
+            indices to class names.
+        @type source_format: str
+        @param source_format: The name of the source format (e.g.,
+            "coco", "imagenet").
+        """
+        ...
+
     @abstractmethod
     def set_skeletons(
         self,

luxonis_ml/data/parsers/fiftyone_classification_parser.py

@@ -164,13 +164,17 @@ def clean_imagenet_annotations(labels_path: Path) -> Path:
 
     This function handles two known issues in ImageNet FiftyOne exports:
 
-    1. Duplicate class names:
-       - First instance of "crane" -> "crane_bird"
-       - Second instance of "maillot" -> "maillot_swim_suit"
+        1. Duplicate class names: First instance of "crane" is renamed
+           to "crane_bird", second instance of "maillot" is renamed to
+           "maillot_swim_suit".
 
-    2. Misindexed labels:
-       - "006742": 517 -> "006742": 134
-       - "031933": 639 -> "031933": 638
+        2. Misindexed labels: "006742" label 517 is corrected to 134,
+           "031933" label 639 is corrected to 638.
+
+    @type labels_path: Path
+    @param labels_path: Path to the labels.json file.
+    @rtype: Path
+    @return: Path to the cleaned labels file.
     """
     with open(labels_path) as f:
         labels_data = json.load(f)