fix draft collection

bioimage-io · Jun 13, 2024 · 76156be · 76156be
1 parent 10ddcdd
commit 76156be
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/bioimageio_collection_backoffice/remote_collection.py b/bioimageio_collection_backoffice/remote_collection.py
@@ -365,10 +365,13 @@ def generate_collection_json(
         id_map: Dict[str, IdInfo] = {}
         for rc in self.get_concepts():
             versions: Union[List[RecordDraft], List[Record]] = (
-                [rc.draft]
-                if mode == "draft" and rc.draft.exists()
+                ([rc.draft] if rc.draft.exists() else [])
+                if mode == "draft"
                 else rc.get_published_versions()
             )
+            if not versions:
+                continue
+
             try:
                 versions_in_collection, id_map_update = create_collection_entries(
                     versions