espnet3.contract.stages.stage_spec

source

espnet3.contract.stages.stage_spec(system_cls: type, name: str) → StageSpec

Return the declared StageSpec named name on system_cls.

• Raises:StageContractError – No stage named name is declared.

Examples

>>> class ExampleSystem:
...     stages = (
...         StageSpec(name="train", config="training", log_dir="exp_dir"),
...     )
>>> stage_spec(ExampleSystem, "train").config
'training'