Skip to content

Commit 0832331

Browse files
committed
Fix xli
1 parent cafa90d commit 0832331

File tree

1 file changed

+7
-4
lines changed

1 file changed

+7
-4
lines changed

src/superannotate/lib/app/interface/cli_interface.py

Lines changed: 7 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -11,7 +11,6 @@
1111
from pathlib import Path
1212
from typing import Any
1313
from typing import Optional
14-
from tqdm import tqdm
1514

1615
import lib.core as constances
1716
import pandas as pd
@@ -92,7 +91,7 @@ def upload_images(
9291
project: str,
9392
folder: str,
9493
extensions: str = constances.DEFAULT_IMAGE_EXTENSIONS,
95-
set_annotation_status: str = constances.AnnotationStatus.NOT_STARTED.value,
94+
set_annotation_status: str = constances.AnnotationStatus.NOT_STARTED.name,
9695
exclude_file_patterns=constances.DEFAULT_FILE_EXCLUDE_PATTERNS,
9796
recursive_subfolders=False,
9897
image_quality_in_editor=None,
@@ -130,7 +129,11 @@ def upload_image(image_path: str):
130129
return ProcessedImage(uploaded=False, path=image_path, entity=None)
131130

132131
paths = []
133-
for extension in extensions.strip().split(","):
132+
133+
if isinstance(extensions,str):
134+
extensions = extensions.strip().split(",")
135+
136+
for extension in extensions:
134137
if recursive_subfolders:
135138
paths += list(Path(folder).rglob(f"*.{extension.lower()}"))
136139
if os.name != "nt":
@@ -165,7 +168,7 @@ def upload_image(image_path: str):
165168
uploaded_image_entities.append(processed_image.entity)
166169
else:
167170
failed_images.append(processed_image.path)
168-
progress_bar.update(1)
171+
progress_bar.update(1)
169172

170173
for i in range(0, len(uploaded_image_entities), 500):
171174
self.controller.upload_images(

0 commit comments

Comments
 (0)