fix: uncomment

luxonis · Nov 4, 2024 · 90d444c · 90d444c
1 parent 30d4bbf
commit 90d444c
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/luxonis_ml/data/datasets/luxonis_dataset.py b/luxonis_ml/data/datasets/luxonis_dataset.py
@@ -310,8 +310,10 @@ def get_classes(self) -> Tuple[List[str], Dict[str, List[str]]]:
                 for c in classes
             }
         )
-        # for task in self.metadata["classes"]:
-        #     self.metadata["classes"][task] = sorted(self.metadata["classes"][task])
+        for task in self.metadata["classes"]:
+            self.metadata["classes"][task] = sorted(
+                self.metadata["classes"][task]
+            )
         return sorted(all_classes), self.metadata["classes"]
 
     def set_skeletons(