Skip to content

Commit 3139b1b

Browse files
committed
Fix project type error
1 parent 9869059 commit 3139b1b

File tree

4 files changed

+10
-10
lines changed

4 files changed

+10
-10
lines changed

superannotate/db/images.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -660,7 +660,7 @@ def download_image(
660660

661661
project, project_folder = get_project_and_folder_metadata(project)
662662
upload_state = common.upload_state_int_to_str(project.get("upload_state"))
663-
project_type = common.project_type_int_to_str(project['type'])
663+
project_type = project['type']
664664
if upload_state == "External":
665665
raise SABaseException(
666666
0,
@@ -737,7 +737,7 @@ def get_image_bytes(project, image_name, variant='original'):
737737
"""
738738
project, project_folder = get_project_and_folder_metadata(project)
739739
upload_state = common.upload_state_int_to_str(project.get("upload_state"))
740-
project_type = common.project_type_int_to_str(project['type'])
740+
project_type = project['type']
741741

742742
if upload_state == "External":
743743
raise SABaseException(

superannotate/db/project_images.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -58,7 +58,7 @@ def upload_image_to_project(
5858
initial_project_inp = project
5959
project, folder = get_project_and_folder_metadata(project)
6060
upload_state = common.upload_state_int_to_str(project.get("upload_state"))
61-
project_type = common.project_type_int_to_str(project["type"])
61+
project_type = project["type"]
6262
if upload_state == "External":
6363
raise SABaseException(
6464
0,

superannotate/db/projects.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -418,7 +418,7 @@ def upload_video_to_project(
418418
limit = _get_available_image_counts(project, folder)
419419

420420
upload_state = common.upload_state_int_to_str(project.get("upload_state"))
421-
project_type = common.project_type_int_to_str(project.get("type"))
421+
project_type = project.get("type")
422422

423423
if upload_state == "External":
424424
raise SABaseException(
@@ -520,7 +520,7 @@ def upload_videos_from_folder_to_project(
520520
"""
521521
project, folder = get_project_and_folder_metadata(project)
522522
upload_state = common.upload_state_int_to_str(project.get("upload_state"))
523-
project_type = common.project_type_int_to_str(project.get("type"))
523+
project_type = project.get("type")
524524

525525
if upload_state == "External":
526526
raise SABaseException(
@@ -833,7 +833,7 @@ def upload_images_from_public_urls_to_project(
833833
path_to_url = {}
834834
project, project_folder = get_project_and_folder_metadata(project)
835835
upload_state = common.upload_state_int_to_str(project.get("upload_state"))
836-
project_type = common.upload_state_int_to_str(project.get('type'))
836+
project_type = project.get('type')
837837
if upload_state == "External":
838838
raise SABaseException(
839839
0,
@@ -943,7 +943,7 @@ def upload_images_from_google_cloud_to_project(
943943
path_to_url = {}
944944
project, project_folder = get_project_and_folder_metadata(project)
945945
upload_state = common.upload_state_int_to_str(project.get("upload_state"))
946-
project_type = common.project_type_int_to_str(project.get("type"))
946+
project_type = project.get("type")
947947

948948
if upload_state == "External":
949949
raise SABaseException(
@@ -1029,7 +1029,7 @@ def upload_images_from_azure_blob_to_project(
10291029
path_to_url = {}
10301030
project, project_folder = get_project_and_folder_metadata(project)
10311031
upload_state = common.upload_state_int_to_str(project.get("upload_state"))
1032-
project_type = common.project_type_int_to_str(project.get('type'))
1032+
project_type = project.get('type')
10331033
if upload_state == "External":
10341034
raise SABaseException(
10351035
0,

superannotate/ml/ml_funcs.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -61,7 +61,7 @@ def run_prediction(project, images_list, model):
6161
)
6262
project_id = project["id"]
6363
upload_state = upload_state_int_to_str(project.get("upload_state"))
64-
project_type = project_type_str_to_int(project.get("type"))
64+
project_type = project.get("type")
6565
if upload_state == "External":
6666
raise SABaseException(
6767
0,
@@ -138,7 +138,7 @@ def run_segmentation(project, images_list, model):
138138
raise SABaseException(0, "Model Does not exist")
139139

140140
upload_state = upload_state_int_to_str(project.get("upload_state"))
141-
project_type = project_type_str_to_int(project.get("type"))
141+
project_type = project.get("type")
142142
if upload_state == "External":
143143
raise SABaseException(
144144
0,

0 commit comments

Comments
 (0)