tfx.orchestration.portable.base_driver_operator.BaseDriverOperator

View source on GitHub

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.