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

tfx.orchestration.portable.python_driver_operator.PythonDriverOperator

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
)

driver_spec The specification of how to initialize the driver.
mlmd_connection ML metadata connection.

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.

SUPPORTED_EXECUTABLE_SPEC_TYPE [<class 'tfx.proto.orchestration.executable_spec_pb2.PythonClassExecutableSpec'>]