From 7e69a79b3968e53977669459d843b9a28c80b569 Mon Sep 17 00:00:00 2001 From: Dillon Laird Date: Fri, 11 Oct 2024 17:36:39 -0700 Subject: [PATCH] make fine tune id test looser --- tests/integ/test_tools.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/integ/test_tools.py b/tests/integ/test_tools.py index 62a0a771..93680398 100644 --- a/tests/integ/test_tools.py +++ b/tests/integ/test_tools.py @@ -163,7 +163,7 @@ def test_florence2_phrase_grounding_video_fine_tune_id(): fine_tune_id=FINE_TUNE_ID, ) assert len(result) == 10 - assert 16 <= len([res["label"] for res in result[0]]) <= 26 + assert 12 <= len([res["label"] for res in result[0]]) <= 26 assert all([all([0 <= x <= 1 for x in obj["bbox"]]) for obj in result[0]])