Revert change

landing-ai · May 13, 2024 · f52282e · f52282e
1 parent 62311ed
commit f52282e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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