fixed loading annotations

luxonis · Sep 3, 2024 · a561a6d · a561a6d
1 parent c919800
commit a561a6d
Showing 1 changed file with 7 additions and 7 deletions.
diff --git a/luxonis_ml/data/loaders/luxonis_loader.py b/luxonis_ml/data/loaders/luxonis_loader.py
@@ -234,13 +234,13 @@ def _load_image_with_annotations(self, idx: int) -> Tuple[np.ndarray, Labels]:
             data = json.loads(ann_str)
             if type_ == "ArrayAnnotation" and self.dataset.is_remote:
                 data["path"] = self.dataset.arrays_path / data["path"]
-                data.update(
-                    {
-                        "class": class_,
-                        "task": task,
-                        "instance_id": instance_id,
-                    }
-                )
+            data.update(
+                {
+                    "class": class_,
+                    "task": task,
+                    "instance_id": instance_id,
+                }
+            )
             annotation = load_annotation(type_, data)
             labels_by_task[task].append(annotation)