diff --git a/tests/test_tools.py b/tests/test_tools.py index c7d33acc..b35cd7e3 100644 --- a/tests/test_tools.py +++ b/tests/test_tools.py @@ -10,7 +10,7 @@ def test_grounding_dino(): prompt="coin", image=img, ) - assert [l.strip() for l in result["labels"]] == ["coin"] * 24 + assert result["labels"] == ["coin"] * 24 assert len(result["bboxes"]) == 24 assert len(result["scores"]) == 24