runner

Execute and aggregate one compiled post-MIP node.

Functions

aggregate_post_mip_node
expected_post_mip_execution_identity	Return the producer identity expected for the current post-MIP inputs.
register_downstream_evaluation_profile	Register an examples-layer evaluator without coupling ModelOpt to that example.
run_post_mip_node_shard

aggregate_post_mip_node(config, stage_id)

Parameters:

• config (dict[str, Any])

• stage_id (str)

Return type:

dict[str, Any]

expected_post_mip_execution_identity(config, stage_id)

Return the producer identity expected for the current post-MIP inputs.

Parameters:

• config (Mapping[str, Any])

• stage_id (str)

Return type:

str

register_downstream_evaluation_profile(name, evaluator)

Register an examples-layer evaluator without coupling ModelOpt to that example.

Parameters:

• name (str)

• evaluator (Callable[[...], dict[str, Any]])

Return type:

None

run_post_mip_node_shard(config, stage_id, *, shard_index=0, shard_count=1)

Parameters:

• config (dict[str, Any])

• stage_id (str)

• shard_index (int)

• shard_count (int)

Return type:

Path