detections __getitem__ test: minor class name correction

roboflow · Dec 4, 2024 · 7141f11 · 7141f11
1 parent 37eadfe
commit 7141f11
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/test/detection/test_core.py b/test/detection/test_core.py
@@ -225,10 +225,10 @@
         ),
         (
             Detections.empty(),
-            np.isin(Detections.empty()["class_name"], [0, 1, 2]),
+            np.isin(Detections.empty()["class_name"], ["cat", "dog"]),
             Detections.empty(),
             DoesNotRaise(),
-        ),
+        ),  # Filter an empty detections by specific class names
     ],
 )
 def test_getitem(