Skip to content

Commit 5620618

Browse files
committed
Fix tqdm
1 parent f986592 commit 5620618

File tree

2 files changed

+7
-3
lines changed

2 files changed

+7
-3
lines changed

src/superannotate/lib/core/reporter.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -90,7 +90,8 @@ def __enter__(self):
9090
return self
9191

9292
def __exit__(self, type, value, traceback):
93-
self._progress_bar.clos()
93+
if self._progress_bar:
94+
self._progress_bar.close()
9495

9596
def update(self, value=1):
9697
if not self._progress_bar:

src/superannotate/lib/core/usecases/images.py

Lines changed: 5 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3237,8 +3237,8 @@ def execute(self) -> Response:
32373237
if set(duplicate_images) == set(frame_names):
32383238
continue
32393239
uploaded_paths = []
3240-
for _ in frames_generator:
3241-
with Progress(total_frames_count, f"Uploading {Path(path).name}") as progress:
3240+
with Progress(total_frames_count, f"Uploading {Path(path).name}") as progress:
3241+
for _ in frames_generator:
32423242
use_case = UploadImagesFromFolderToProject(
32433243
project=self._project,
32443244
folder=self._folder,
@@ -3269,6 +3269,9 @@ def execute(self) -> Response:
32693269
os.remove(image_path)
32703270
else:
32713271
raise AppException(use_case.response.errors)
3272+
3273+
3274+
32723275
data[str(path)] = uploaded_paths
32733276
self._response.data = data
32743277
return self._response

0 commit comments

Comments
 (0)