Skip to content

Commit

Permalink
add status to draft collection entries
Browse files Browse the repository at this point in the history
  • Loading branch information
FynnBe committed Nov 13, 2024
1 parent d8e6b15 commit b349b91
Show file tree
Hide file tree
Showing 2 changed files with 10 additions and 1 deletion.
3 changes: 3 additions & 0 deletions bioimageio_collection_backoffice/collection_json.py
Original file line number Diff line number Diff line change
Expand Up @@ -11,6 +11,7 @@
CollectionWebsiteConfigTemplate,
)
from .common import Node
from .db_structure.version_info import DraftStatus, ErrorStatus


class Author(Node, frozen=True):
Expand Down Expand Up @@ -58,6 +59,8 @@ class CollectionEntry(Node, frozen=True):
training_data: Optional[TrainingData] = None
type: Literal["application", "model", "notebook", "dataset"]
source: Optional[str] = None
status: Optional[Union[DraftStatus, ErrorStatus]] = None
"""status of the draft (for collection_draft.json only)"""

def __lt__(self, other: CollectionEntry):
sdc = 0 if self.download_count == "?" else self.download_count
Expand Down
8 changes: 7 additions & 1 deletion bioimageio_collection_backoffice/remote_collection.py
Original file line number Diff line number Diff line change
Expand Up @@ -37,7 +37,7 @@
from loguru import logger
from pydantic import AnyUrl
from ruyaml import YAML
from typing_extensions import Concatenate, ParamSpec
from typing_extensions import Concatenate, ParamSpec, assert_never

from ._settings import settings
from ._thumbnails import create_thumbnails
Expand Down Expand Up @@ -1177,6 +1177,11 @@ def create_collection_entries(
f"{record_version.concept.folder}versions.json",
versions_info.model_dump(mode="json"),
)
status = None
elif isinstance(record_version, RecordDraft):
status = record_version.info.status
else:
assert_never(record_version)

try:
# legacy nickname
Expand Down Expand Up @@ -1274,5 +1279,6 @@ def get_compat_tag(tool: str):
training_data=rdf["training_data"] if "training_data" in rdf else None,
type=rdf["type"],
source=rdf.get("source"),
status=status,
)
], id_map

0 comments on commit b349b91

Please sign in to comment.