fix: honor limit parameter in all cases in run-object-detection CLI

openfoodfacts · Dec 28, 2022 · 4268939 · 4268939
1 parent 4420346
commit 4268939
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/robotoff/cli/main.py b/robotoff/cli/main.py
@@ -380,6 +380,9 @@ def run_object_detection_model(
             if not parsed_url.netloc or not parsed_url.scheme:
                 raise ValueError(f"invalid image URL: {image_url}")
 
+        if limit:
+            image_urls = image_urls[:limit]
+
     else:
         with db:
             query = (