| 123456789101112131415161718192021222324252627 |
- from typing import Any, Dict, Literal, final
- from wandb._pydantic import field_validator, to_camel
- from wandb.sdk.artifacts.artifact_manifest_entry import ArtifactManifestEntry
- from .base_model import ArtifactsBase
- from .storage import StoragePolicyConfig
- @final
- class ArtifactManifestV1Data(ArtifactsBase, alias_generator=to_camel):
- """Data model for the v1 artifact manifest."""
- version: Literal[1]
- contents: Dict[str, ArtifactManifestEntry]
- storage_policy: str
- storage_policy_config: StoragePolicyConfig
- @field_validator("contents", mode="before")
- def _validate_entries(cls, v: Any) -> Any:
- # The dict keys should be the `entry.path` values, but they've
- # historically been dropped from the JSON objects. This restores
- # them on instantiation.
- # Pydantic will handle converting dicts -> ArtifactManifestEntries.
- return {path: {**dict(entry), "path": path} for path, entry in v.items()}
|
