From f036774765723b9b1a5077225352850619f5cf2b Mon Sep 17 00:00:00 2001 From: Jeremy Stanley Date: Mon, 15 Jan 2024 10:20:29 -0700 Subject: [PATCH] deal with missing storageFilename --- app/lib/duplicate_image_detector.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/app/lib/duplicate_image_detector.py b/app/lib/duplicate_image_detector.py index a2e0d64..5f041fc 100644 --- a/app/lib/duplicate_image_detector.py +++ b/app/lib/duplicate_image_detector.py @@ -110,12 +110,12 @@ def _calculate_embeddings(self): with ImageEmbedder.create_from_options(options) as embedder: for i in trange(len(self.media_items), ascii=False): media_item = self.media_items[i] - storage_path = self._get_storage_path(media_item) mp_image = None try: + storage_path = self._get_storage_path(media_item) mp_image = mp.Image.create_from_file(storage_path) - except RuntimeError as error: + except (KeyError, RuntimeError) as error: logging.warning( f"Skipping invalid image file:\n" f"error: {error}\n"