Skip to content

Commit 17c933d

Browse files
committed
Fix image unassign
1 parent 3c2fef9 commit 17c933d

File tree

2 files changed

+24
-3
lines changed

2 files changed

+24
-3
lines changed

superannotate/db/project_images.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -688,14 +688,13 @@ def unassign_images(project, image_names):
688688
:param image_names: list of image unassign
689689
:type image_names: list of str
690690
"""
691-
project_meta = get_project_metadata(project)
692691
project, folder = get_project_and_folder_metadata(project)
693692
folder_name = 'root'
694693
if folder:
695694
folder_name = folder['name']
696695
params = {
697-
"project_id": project_meta['id'],
698-
"team_id": project_meta["team_id"]
696+
"project_id": project['id'],
697+
"team_id": project["team_id"]
699698
}
700699
json_req = {
701700
"image_names": image_names,

tests/test_assign_images.py

Lines changed: 22 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -147,6 +147,28 @@ def test_unassign_images(tmpdir):
147147

148148
assert im1_metadata["qa_id"] == None
149149
assert im2_metadata["qa_id"] == None
150+
FOLDER_NAME = "test_folder"
151+
sa.create_folder(project, FOLDER_NAME)
152+
project = PROJECT_NAME_VECTOR1 + "/" + FOLDER_NAME
153+
sa.move_images(
154+
PROJECT_NAME_VECTOR1, ["example_image_1.jpg", "example_image_2.jpg"], project
155+
)
156+
sa.assign_images(
157+
project, ["example_image_1.jpg", "example_image_2.jpg"], email
158+
)
159+
time.sleep(1)
160+
sa.unassign_images(
161+
project,
162+
["example_image_1.jpg", "example_image_2.jpg"],
163+
)
164+
165+
sa.search_images(project)
166+
im1_metadata = sa.get_image_metadata(project, "example_image_1.jpg")
167+
168+
im2_metadata = sa.get_image_metadata(project, "example_image_2.jpg")
169+
170+
assert im1_metadata["qa_id"] == None
171+
assert im2_metadata["qa_id"] == None
150172

151173

152174
def test_assign_folder(tmpdir):

0 commit comments

Comments
 (0)