From c870480c21bdcd0dfd767ca4e8c9f97e5ec2e535 Mon Sep 17 00:00:00 2001 From: Dillon Laird Date: Tue, 12 Mar 2024 17:58:51 -0700 Subject: [PATCH] fix import --- tests/test_llm.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/tests/test_llm.py b/tests/test_llm.py index 1dd39dcb..a8070f30 100644 --- 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