diff --git a/bioimageio_collection_backoffice/collection_json.py b/bioimageio_collection_backoffice/collection_json.py index 109b4868..deffcf2d 100644 --- 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 index 15114a19..4d2a212f 100644 --- 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