The base class of all executor operators.
tfx.orchestration.portable.base_driver_operator.BaseDriverOperator(
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
SUPPORTED_EXECUTABLE_SPEC_TYPE
View source
SUPPORTED_EXECUTABLE_SPEC_TYPE()
run_driver
View source
@abc.abstractmethod
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.
|