fix: fix ignore_extra_sizes method

openfoodfacts · Nov 14, 2024 · bcc0e9b · bcc0e9b
1 parent 9dd2530
commit bcc0e9b
Showing 1 changed file with 4 additions and 3 deletions.
diff --git a/openfoodfacts_exports/exports/parquet.py b/openfoodfacts_exports/exports/parquet.py
@@ -235,9 +235,10 @@ def ignore_extra_sizes(self):
         """Literal doesn't accept extra values, returning an error in case of
         additional keys.
         """
-        self.sizes = {
-            k: v for k, v in self.sizes.items() if k in ALLOWED_IMAGE_SIZE_KEYS
-        }
+        if self.sizes:
+            self.sizes = {
+                k: v for k, v in self.sizes.items() if k in ALLOWED_IMAGE_SIZE_KEYS
+            }
         return self
 
     @model_validator(mode="before")