From 5340e464a190ad8977817c7c1249fb78a24b4cb4 Mon Sep 17 00:00:00 2001 From: Dillon Laird Date: Tue, 7 May 2024 15:51:14 -0700 Subject: [PATCH] fixed test cases --- tests/test_type_defs.py | 2 +- tests/tools/test_video.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/tests/test_type_defs.py b/tests/test_type_defs.py index 273681f6..eafb670e 100644 --- a/tests/test_type_defs.py +++ b/tests/test_type_defs.py @@ -1,6 +1,6 @@ import os -from vision_agent.type_defs import LandingaiAPIKey +from vision_agent.utils.type_defs import LandingaiAPIKey def test_load_api_credential_from_env_var(): diff --git a/tests/tools/test_video.py b/tests/tools/test_video.py index a19b529f..f77ae2df 100644 --- a/tests/tools/test_video.py +++ b/tests/tools/test_video.py @@ -1,4 +1,4 @@ -from vision_agent.tools.video import extract_frames_from_video +from vision_agent.utils.video import extract_frames_from_video def test_extract_frames_from_video():