jaxdem.utils.serialization_schema#

Shared file-level serialization schema for Orbax and HDF5 payloads.

Functions

make_serialization_manifest(storage, payload)

Describe a trusted JaxDEM payload independently of its storage format.

serialization_schema_version(manifest, *[, ...])

Validate a manifest, or accept an explicitly supported legacy version.

jaxdem.utils.serialization_schema.make_serialization_manifest(storage: str, payload: str) → dict[str, Any][source]#

Describe a trusted JaxDEM payload independently of its storage format.

jaxdem.utils.serialization_schema.serialization_schema_version(manifest: Any, *, legacy_version: Any = 1, expected_storage: str) → int[source]#

Validate a manifest, or accept an explicitly supported legacy version.