Merge pull request #367 from openfoodfacts/export-ls-to-hf-was-cancelled

feat: 🎨 Consider "was_cancelled" during export from ls to hf
openfoodfacts · Dec 4, 2024 · 7bd75a3 · 7bd75a3
2 parents b8b3a1a + c1b8f8a
commit 7bd75a3
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/ml_utils/ml_utils_cli/cli/sample.py b/ml_utils/ml_utils_cli/cli/sample.py
@@ -121,6 +121,10 @@ def format_object_detection_sample_to_hf(
         return None
 
     annotation = annotations[0]
+    if annotation["was_cancelled"]:
+        logger.debug("Annotation was cancelled, skipping.")
+        return None
+
     bboxes = []
     bbox_category_ids = []
     bbox_category_names = []