Join the SIG TFX-Addons community and help make TFX even better!

tfx.orchestration.experimental.core.pipeline_ops.stop_node

Stops a node in an async pipeline.

tfx.orchestration.experimental.core.pipeline_ops.stop_node(
    mlmd_handle: tfx.orchestration.metadata.Metadata,
    node_uid: tfx.orchestration.experimental.core.task.NodeUid,
    timeout_secs: float = DEFAULT_WAIT_FOR_INACTIVATION_TIMEOUT_SECS
) -> None

Initiates a node stop operation and waits for the node execution to become inactive.

mlmd_handle A handle to the MLMD db.
node_uid Uid of the node to be stopped.
timeout_secs Amount of time in seconds to wait for node to stop.

status_lib.StatusNotOkError Failure to stop the node.