From 7141f11c0b2a738fb24f0c99d044df55c34df63a Mon Sep 17 00:00:00 2001 From: LinasKo Date: Wed, 4 Dec 2024 12:13:33 +0200 Subject: [PATCH] detections __getitem__ test: minor class name correction --- test/detection/test_core.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/test/detection/test_core.py b/test/detection/test_core.py index 402d9ef1c..dfa784fcb 100644 --- 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(