Remove assets in export without labels (#1787)

kili-technology · Oct 9, 2024 · 89e2468 · 89e2468
1 parent 039ee6e
commit 89e2468
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/kili/services/export/tools.py b/src/kili/services/export/tools.py
@@ -168,10 +168,12 @@ def fetch_assets(
 
     if (label_type_in is not None) and (len(label_type_in) > 0):
         if export_type == "latest":
+            assets_gen = filter(lambda asset: asset.get("latestLabel") is not None, assets_gen)
             assets_gen = filter(
                 lambda asset: asset["latestLabel"].get("labelType") in label_type_in, assets_gen
             )
         else:
+            assets_gen = filter(lambda asset: asset.get("labels") is not None, assets_gen)
             assets_gen = filter(
                 lambda asset: any(
                     label.get("labelType") in label_type_in for label in asset["labels"]