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.
|