fix: add with_db decorator to generate_fiber_quality_facet

openfoodfacts · Dec 9, 2022 · c62f14c · c62f14c
1 parent 3e1ec39
commit c62f14c
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/robotoff/scheduler/latent.py b/robotoff/scheduler/latent.py
@@ -4,7 +4,7 @@
 
 from robotoff import settings
 from robotoff.insights.dataclass import InsightType
-from robotoff.models import Prediction, ProductInsight
+from robotoff.models import Prediction, ProductInsight, with_db
 from robotoff.off import get_server_type
 from robotoff.prediction.types import PredictionType
 from robotoff.products import (
@@ -30,6 +30,7 @@ def generate_quality_facets():
     generate_fiber_quality_facet()
 
 
+@with_db
 def generate_fiber_quality_facet():
     product_store: DBProductStore = get_product_store()
     collection = product_store.collection