Skip to content

Orchestrate: Config

Boundary types shared by the Typer CLI and the stage primitives. ResolvedConfig is the handoff from render + validate_config into build / train / evaluate; InstantiatedRun is the wired (trainer, model, datamodule) triple produced by build_run.

graphids.orchestrate.config

config

Orchestrate data types.

Boundary types shared by the CLI and the stage primitives. ResolvedConfig is the handoff from render → validate into build / train / evaluate; InstantiatedRun is the wired (trainer, model, datamodule) triple produced by instantiate.

InstantiatedRun dataclass 

InstantiatedRun(trainer: Trainer, model: Module, datamodule: Any)

A wired (trainer, model, datamodule) triple built from a rendered config.

ResolvedConfig dataclass 

ResolvedConfig(rendered: dict[str, Any], validated: ValidatedConfig, stage_name: str, run_dir: Path | None, ckpt_file: Path | None)

Rendered, validated config ready for instantiation.

Stage primitives (orchestrate/stage.py) consume this directly: they read rendered + validated for building trainer/model, run_dir / ckpt_file for marker + OTel export wiring, and stage_name for log fields. run_dir is None only for smoke invocations of the Typer CLI with no default_root_dir set — markers and file exporters are skipped in that case.

from_rendered classmethod 

from_rendered(rendered: dict[str, Any], *, stage_name: str) -> ResolvedConfig

Validate a pre-rendered dict and pull run_dir from jsonnet.

Source code in graphids/orchestrate/config.py 
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
@classmethod
def from_rendered(cls, rendered: dict[str, Any], *, stage_name: str) -> ResolvedConfig:
    """Validate a pre-rendered dict and pull ``run_dir`` from jsonnet."""
    from graphids.config.constants import CKPT_SUBPATH
    from graphids.config.schemas import validate_config

    validated = validate_config(rendered)
    default_root = (rendered.get("trainer") or {}).get("default_root_dir") or ""
    run_dir = Path(default_root) if default_root else None
    ckpt_file = run_dir / CKPT_SUBPATH if run_dir else None
    return cls(
        rendered=rendered,
        validated=validated,
        stage_name=stage_name,
        run_dir=run_dir,
        ckpt_file=ckpt_file,
    )