fixed test cases

landing-ai · May 7, 2024 · 5340e46 · 5340e46
1 parent a3b435c
commit 5340e46
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git 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
@@ -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():