nerfbaselines.registry

class nerfbaselines.registry.DatasetSpec

Bases: TypedDict

download_dataset_function: str

evaluation_protocol: str | EvaluationProtocolWithNameSpec

load_dataset_function: Required[str]

metadata: DatasetSpecMetadata

priority: Required[int]

class nerfbaselines.registry.EvaluationProtocolSpec

Bases: TypedDict

evaluation_protocol: Required[str]

class nerfbaselines.registry.EvaluationProtocolWithNameSpec

Bases: EvaluationProtocolSpec

evaluation_protocol: Required[str]

name: Required[str]

class nerfbaselines.registry.MethodSpec

Bases: TypedDict

apptainer: NotRequired[dict]

backends_order: List[str]

conda: NotRequired[dict]

dataset_overrides: Dict[str, Any]

docker: NotRequired[dict]

kwargs: Dict[str, Any]

metadata: Dict[str, Any]

method: Required[str]

nerfbaselines.registry.assert_not_none(value: T | None) → T

nerfbaselines.registry.build_evaluation_protocol(name: str) → EvaluationProtocol

nerfbaselines.registry.build_method(method: str, backend: str | None = None)

nerfbaselines.registry.get_dataset_downloaders() → Dict[str, DownloadDatasetFunction]

nerfbaselines.registry.get_dataset_loaders() → Sequence[Tuple[str, LoadDatasetFunction]]

nerfbaselines.registry.get_dataset_overrides(method_name: str, dataset_metadata: Dict[str, Any]) → Dict[str, Any] | None

nerfbaselines.registry.get_dataset_spec(name: str) → DatasetSpec

nerfbaselines.registry.get_method_spec(name: str) → MethodSpec

Get a method by name

nerfbaselines.registry.get_supported_datasets() → FrozenSet[str]

nerfbaselines.registry.get_supported_methods(backend_name: str | None = None) → FrozenSet[str]

nerfbaselines.registry.partialmethod(cls, method_name, **kwargs)

nerfbaselines.registry.register(spec: MethodSpec | DatasetSpec | EvaluationProtocolSpec, *, name: str, metadata=None, kwargs=None, dataset_overrides=None) → None

nerfbaselines.registry.register_logger(name: str, logger: Callable[[...], Logger]) → None