fixed bad test case

landing-ai · Mar 13, 2024 · e50dacd · e50dacd
1 parent 8ccada3
commit e50dacd
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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"