Questions about TFX? Join us at Google I/O!

tfx.orchestration.portable.base_driver.BaseDriver

The base class of all drivers.

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

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.