fixed task name for native parser

luxonis · Jan 14, 2025 · 8de4ad0 · 8de4ad0
1 parent 2df74e6
commit 8de4ad0
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/luxonis_ml/data/parsers/base_parser.py b/luxonis_ml/data/parsers/base_parser.py
@@ -243,7 +243,8 @@ def _add_task(self, generator: DatasetIterator) -> DatasetIterator:
         task_name = self.task_name or ""
         for item in generator:
             if isinstance(item, dict):
-                item["task"] = task_name
-            else:
+                if "task" not in item:
+                    item["task"] = task_name
+            elif not item.task:
                 item.task = task_name
             yield item