fix import

landing-ai · Mar 13, 2024 · c870480 · c870480
1 parent 3abfb50
commit c870480
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/tests/test_llm.py b/tests/test_llm.py
@@ -1,8 +1,7 @@
 import pytest
 
 from vision_agent.llm.llm import OpenAILLM
-from vision_agent.tools import CLIP
-from vision_agent.tools.tools import GroundingDINO
+from vision_agent.tools import CLIP, GroundingDINO, GroundingSAM
 
 from .fixtures import openai_llm_mock # noqa: F401