fix: fix issue with max() call on empty sequence

openfoodfacts · Oct 25, 2024 · 5410b30 · 5410b30
1 parent 253d0b7
commit 5410b30
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/robotoff/workers/tasks/import_image.py b/robotoff/workers/tasks/import_image.py
@@ -842,14 +842,15 @@ def extract_nutrition_job(
             return
 
         output = nutrition_extraction.predict(image, ocr_result, triton_uri=triton_uri)
+        max_confidence = None
 
         if output is None:
             data: JSONType = {"error": "missing_text"}
-            max_confidence = None
         else:
-            max_confidence = max(
-                entity["score"] for entity in output.entities.aggregated
-            )
+            if output.entities.aggregated:
+                max_confidence = max(
+                    entity["score"] for entity in output.entities.aggregated
+                )
             data = {
                 "nutrients": {
                     entity: dataclasses.asdict(nutrient)