prervent division by zero error

zauberzeug · Aug 30, 2024 · e561789 · e561789
1 parent 7f1e1c9
commit e561789
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/learning_loop_node/data_exchanger.py b/learning_loop_node/data_exchanger.py
@@ -101,6 +101,12 @@ async def download_images(self, image_uuids: List[str], image_folder: str, chunk
 
         paths, _ = create_resource_paths(self.context.organization, self.context.project, new_image_uuids)
         num_new_image_ids = len(new_image_uuids)
+        if num_new_image_ids == 0:
+            logging.info('All images are already downloaded. Nothing to do.')
+            self.progress = 1.0
+            return
+
+        logging.info(f'Downloading {num_new_image_ids} new images to {image_folder}..')
         os.makedirs(image_folder, exist_ok=True)
 
         progress_factor = 0.5 / num_new_image_ids  # second 50% of progress is for downloading images