petitRADTRANS.sbi.estimator_registry

Contents

petitRADTRANS.sbi.estimator_registry#

Registry and generic loader utilities for persisted posterior estimators.

Functions#

`register_posterior_estimator`(→ None)	Register one estimator family for generic load dispatch.
`get_posterior_estimator_class`(...)	Return the registered estimator class for one estimator family.
`read_posterior_metadata`(→ dict[str, Any])	Read the persisted metadata envelope for one saved estimator.
`resolve_estimator_family_from_metadata`(→ str)	Infer the estimator family from current or legacy saved metadata.
`load_posterior_estimator`(...)	Load a saved posterior estimator without naming its concrete class.

Module Contents#

petitRADTRANS.sbi.estimator_registry.register_posterior_estimator(estimator_family: str, estimator_cls: type[petitRADTRANS.sbi.posterior.PosteriorEstimator]) → None#: Register one estimator family for generic load dispatch.

petitRADTRANS.sbi.estimator_registry.get_posterior_estimator_class(estimator_family: str) → type[petitRADTRANS.sbi.posterior.PosteriorEstimator]#: Return the registered estimator class for one estimator family.

petitRADTRANS.sbi.estimator_registry.read_posterior_metadata(input_directory: str) → dict[str, Any]#: Read the persisted metadata envelope for one saved estimator.

petitRADTRANS.sbi.estimator_registry.resolve_estimator_family_from_metadata(metadata: dict[str, Any]) → str#: Infer the estimator family from current or legacy saved metadata.

petitRADTRANS.sbi.estimator_registry.load_posterior_estimator(input_directory: str) → petitRADTRANS.sbi.posterior.PosteriorEstimator#: Load a saved posterior estimator without naming its concrete class.