detections __getitem__: earlier check for empty detections

roboflow · Dec 3, 2024 · a4dfcac · a4dfcac
1 parent 28fa054
commit a4dfcac
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/supervision/detection/core.py b/supervision/detection/core.py
@@ -1201,10 +1201,10 @@ def __getitem__(
         """
         if isinstance(index, str):
             return self.data.get(index)
-        if isinstance(index, int):
-            index = [index]
         if self.is_empty():
             return Detections.empty()
+        if isinstance(index, int):
+            index = [index]
         return Detections(
             xyxy=self.xyxy[index],
             mask=self.mask[index] if self.mask is not None else None,