Improve test robustness

landing-ai · May 13, 2024 · 62311ed · 62311ed
1 parent 8bcae34
commit 62311ed
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 result["labels"] == ["coin"] * 24
+ assert [l.strip() for l in result["labels"]] == ["coin"] * 24
  assert len(result["bboxes"]) == 24
  assert len(result["scores"]) == 24