tfx.orchestration.portable.base_driver.BaseDriver

The base class of all drivers.

tfx.orchestration.portable.base_driver.BaseDriver(
    mlmd_connection: tfx.orchestration.metadata.Metadata
)

Args
mlmd_connection	ML metadata connection.

Methods

run

View source

@abc.abstractmethod
run(
    execution_info: tfx.orchestration.portable.data_types.ExecutionInfo
) -> driver_output_pb2.DriverOutput

Invokes the driver with inputs provided by the Launcher.

Args
execution_info	a data_types.ExecutionInfo instance representing the execution info needed for the driver execution.

Returns
An DriverOutput instance.