|View source on GitHub|
Task generator for executing a sync pipeline.
tfx.orchestration.experimental.core.pipeline_state.PipelineState, is_task_id_tracked_fn: Callable[[task_lib.TaskId], bool], service_job_manager:
generate is not thread-safe. Concurrent calls to
be explicitly serialized. Since MLMD may be updated upon call to
it's also not safe to call
generate on different instances of this class
where the instances refer to the same MLMD db and the same pipeline IR.
||A handle to the MLMD db.|
A callable that returns
||Used for handling service nodes in the pipeline.|
generate() -> List[
Generates tasks for executing the next executable nodes in the pipeline.
The returned tasks must have
exec_task populated. List may be empty if
no nodes are ready for execution.