From 62311eddede0c8ef805a924b8954f77c0220b0dd Mon Sep 17 00:00:00 2001
From: Yazhou Cao <cyz19892002@gmail.com>
Date: Mon, 13 May 2024 10:48:25 -0700
Subject: [PATCH] Improve test robustness

---
 tests/test_tools.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/tests/test_tools.py b/tests/test_tools.py
index b35cd7e3..c7d33acc 100644
--- 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