From e50dacd74f438a81abda6f971a09829ef72b150b Mon Sep 17 00:00:00 2001 From: Dillon Laird Date: Tue, 12 Mar 2024 17:50:30 -0700 Subject: [PATCH] fixed bad test case --- tests/test_llm.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/test_llm.py b/tests/test_llm.py index 74453a4b..1dd39dcb 100644 --- a/tests/test_llm.py +++ b/tests/test_llm.py @@ -54,6 +54,6 @@ def test_generate_detector(openai_llm_mock): # noqa: F811 def test_generate_segmentor(openai_llm_mock): # noqa: F811 llm = OpenAILLM() prompt = "Can you generate a cat segmentor?" - segmentor = llm.generate_detector(prompt) - assert isinstance(segmentor, GroundingDINO) + segmentor = llm.generate_segmentor(prompt) + assert isinstance(segmentor, GroundingSAM) assert segmentor.prompt == "cat"