base

Registered post-MIP node contracts and campaign flow compilation.

Classes

CompiledPostMIPNode

NodeCapabilities

NodeKind

PostMIPNode

Functions

compile_post_mip_flows

Compile configured flows into validated single-parent dynamic stages.

post_mip_node

render_post_mip_node_report

Render one node using the report hook owned by its registered class.
class CompiledPostMIPNode

Bases: object

__init__(flow_id, node_id, stage_id, node_type, input_id, parent_stage_id, dependency_stage_ids, model_source, config, capabilities, input_artifact, output_artifact, metric_references)
Parameters:

  • flow_id (str)

  • node_id (str)

  • stage_id (str)

  • node_type (str)

  • input_id (str)

  • parent_stage_id (str)

  • dependency_stage_ids (tuple[str, ...])

  • model_source (str)

  • config (dict[str, Any])

  • capabilities (NodeCapabilities)

  • input_artifact (str)

  • output_artifact (str)

  • metric_references (tuple[str, ...])

Return type:

None

capabilities: NodeCapabilities
config: dict[str, Any]
dependency_stage_ids: tuple[str, ...]
flow_id: str
input_artifact: str
input_id: str
metric_references: tuple[str, ...]
model_source: str
node_id: str
node_type: str
output_artifact: str
parent_stage_id: str
stage_id: str
class NodeCapabilities

Bases: object

__init__(kind, accepts, output=None, distributed=False, implemented=True, default_strategy='single')
Parameters:
Return type:

None

accepts: frozenset[ArtifactKind]
default_strategy: str = 'single'
distributed: bool = False
implemented: bool = True
kind: NodeKind
output: ArtifactKind | None = None
class NodeKind

Bases: str, Enum

EVALUATOR = 'evaluator'
SELECTOR = 'selector'
TRANSFORMER = 'transformer'
__new__(value)
class PostMIPNode

Bases: ABC

capabilities: ClassVar[NodeCapabilities]
common_fields: ClassVar[frozenset[str]] = frozenset({'config', 'failure_policy', 'input', 'model_source', 'type'})
classmethod metric_references(config)
Parameters:

config (Mapping[str, Any])

Return type:

tuple[str, …]

classmethod render_report(node, payload)

Return a report fragment, or None for DAG-only node types.

Parameters:
Return type:

str | None

type_name: ClassVar[str]
classmethod validate_config(config)
Parameters:

config (Mapping[str, Any])

Return type:

None

compile_post_mip_flows(config)

Compile configured flows into validated single-parent dynamic stages.

Parameters:

config (Mapping[str, Any])

Return type:

tuple[CompiledPostMIPNode, …]

post_mip_node(cls)
Parameters:

cls (type[PostMIPNode])

Return type:

type[PostMIPNode]

render_post_mip_node_report(node, payload)

Render one node using the report hook owned by its registered class.

Parameters:
Return type:

str | None