fix: fix get_type call

openfoodfacts · Jul 11, 2024 · 18181bb · 18181bb
1 parent 1e15d72
commit 18181bb
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/robotoff/insights/importer.py b/robotoff/insights/importer.py
@@ -1222,7 +1222,8 @@ def get_nutrition_table_predictions(
             .where(
                 ImageModel.barcode == product_id.barcode,
                 ImageModel.server_type == product_id.server_type.name,
-                ImagePrediction.model_name == ObjectDetectionModel.get_type(),
+                ImagePrediction.model_name
+                == ObjectDetectionModel.nutrition_table_yolo.get_type(),
                 ImagePrediction.max_confidence >= min_score,
             )
             .tuples()