add source field

bioimage-io · Jun 13, 2024 · 633db73 · 633db73
1 parent f809ba5
commit 633db73
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git a/bioimageio_collection_backoffice/collection_json.py b/bioimageio_collection_backoffice/collection_json.py
@@ -51,6 +51,7 @@ class CollectionEntry(Node, frozen=True):
     tags: Sequence[str] = ()
     training_data: Optional[TrainingData] = None
     type: Literal["application", "model", "notebook", "dataset"]
+    source: Optional[str] = None
 
     def __lt__(self, other: CollectionEntry):
         sdc = 0 if self.download_count == "?" else self.download_count

diff --git a/bioimageio_collection_backoffice/remote_collection.py b/bioimageio_collection_backoffice/remote_collection.py
@@ -1204,5 +1204,6 @@ def create_collection_entries(
             tags=rdf.get("tags", []),
             training_data=rdf["training_data"] if "training_data" in rdf else None,
             type=rdf["type"],
+            source=rdf.get("source"),
         )
     ], id_map