Skip to content

Commit f6a30ee

Browse files
committed
Fix video folder upload
1 parent 39efead commit f6a30ee

File tree

2 files changed

+13
-3
lines changed

2 files changed

+13
-3
lines changed

superannotate/db/projects.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -237,8 +237,8 @@ def upload_video_to_project(
237237
:return: filenames of uploaded images
238238
:rtype: list of strs
239239
"""
240-
project_metadata, project_folder = get_project_and_folder_metadata(project)
241-
upload_state = common.upload_state_int_to_str(project_metadata.get("upload_state"))
240+
project, project_folder = get_project_and_folder_metadata(project)
241+
upload_state = common.upload_state_int_to_str(project.get("upload_state"))
242242
if upload_state == "External":
243243
raise SABaseException(
244244
0,
@@ -339,7 +339,7 @@ def upload_video_to_project(
339339
)
340340

341341
filenames = upload_images_from_folder_to_project(
342-
project,
342+
(project, project_folder),
343343
tempdir.name,
344344
extensions=["jpg"],
345345
annotation_status=annotation_status,

tests/test_video.py

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -44,3 +44,13 @@ def test_video(tmpdir):
4444
assert len(sa.search_images(PROJECT_NAME1)) == len(
4545
sa.search_images(PROJECT_NAME2)
4646
)
47+
48+
sa.create_folder(project, "new folder")
49+
50+
sa.upload_videos_from_folder_to_project(
51+
PROJECT_NAME2 + "/new folder", "./tests/sample_videos", target_fps=2
52+
)
53+
54+
assert len(sa.search_images(PROJECT_NAME2 + "/new folder")) == len(
55+
sa.search_images(PROJECT_NAME2)
56+
)

0 commit comments

Comments
 (0)