Stops a node in a 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.
Args |
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.
|
Raises |
status_lib.StatusNotOkError
|
Failure to stop the node.
|