|View source on GitHub|
Interface for task schedulers.
tfx.orchestration.metadata.Metadata, pipeline: pipeline_pb2.Pipeline, task:
||A handle to the MLMD db.|
||The pipeline IR proto.|
||Task to be executed.|
cancel() -> None
Cancels task scheduler.
This method will be invoked from a different thread than the thread that's
blocked on call to
cancel must return immediately when called.
schedule method is expected to stop any ongoing work,
clean up and return as soon as possible.
Schedules task execution and returns the results of execution.
This method blocks until task execution completes (successfully or not) or
until explicitly cancelled by a call to
cancel. When cancelled,
is expected to stop any ongoing work, clean up and return as soon as
possible. Note that
cancel will be invoked from a different thread than
schedule and hence the concrete implementations must be thread safe.