refactor conditional logic

luxonis · Nov 27, 2024 · 3716ba7 · 3716ba7
1 parent b39761a
commit 3716ba7
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/luxonis_ml/data/loaders/luxonis_loader.py b/luxonis_ml/data/loaders/luxonis_loader.py
@@ -330,11 +330,11 @@ def _load_image_with_annotations(
                         (len(self.class_mappings[task]), height, width),
                         dtype=np.uint8,
                     )
-                if task == LabelType.BOUNDINGBOX:
+                elif task == LabelType.BOUNDINGBOX:
                     empty_array = np.zeros((0, 6), dtype=np.float32)
-                if task == LabelType.KEYPOINTS:
+                elif task == LabelType.KEYPOINTS:
                     empty_array = np.zeros((0, 3), dtype=np.float32)
-                if task == LabelType.CLASSIFICATION:
+                elif task == LabelType.CLASSIFICATION:
                     empty_array = np.zeros(
                         (0, len(self.class_mappings[task])), dtype=np.float32
                     )