PythonDriverOperator handles python class based driver's init and execution.
Inherits From: BaseDriverOperator
tfx.orchestration.portable.python_driver_operator.PythonDriverOperator(
driver_spec: message.Message,
mlmd_connection: tfx.orchestration.metadata.Metadata
)
Args |
driver_spec
|
The specification of how to initialize the driver.
|
mlmd_connection
|
ML metadata connection.
|
Raises |
RuntimeError
|
if the driver_spec is not supported.
|
Methods
run_driver
View source
run_driver(
execution_info: tfx.orchestration.portable.data_types.ExecutionInfo
) -> driver_output_pb2.DriverOutput
Invokes the driver with inputs provided by the Launcher.
Args |
execution_info
|
data_types.ExecutionInfo containing information needed for
driver execution.
|
Returns |
An DriverOutput instance.
|
Class Variables |
SUPPORTED_EXECUTABLE_SPEC_TYPE
|
|