tfx.orchestration.portable.python_driver_operator.PythonDriverOperator

PythonDriverOperator handles python class based driver's init and execution.

Inherits From: BaseDriverOperator

tfx.orchestration.portable.python_driver_operator.PythonDriverOperator(
    driver_spec: message.Message,
    mlmd_connection: tfx.orchestration.metadata.Metadata,
    pipeline_info: pipeline_pb2.PipelineInfo,
    pipeline_node: pipeline_pb2.PipelineNode
)

driver_spec The specification of how to initialize the driver.
mlmd_connection ML metadata connection.
pipeline_info The information of the pipeline that this driver is in.
pipeline_node The specification of the node that this driver is in.

RuntimeError if the driver_spec is not supported.

Methods

run_driver

View source

run_driver(
    input_dict: Dict[Text, List[tfx.types.Artifact]],
    output_dict: Dict[Text, List[tfx.types.Artifact]],
    exec_properties: Dict[Text, Any]
) -> driver_output_pb2.DriverOutput

Invokes the driver with inputs provided by the Launcher.

Args
input_dict The defult input_dict resolved by the launcher.
output_dict The default output_dict resolved by the launcher.
exec_properties The default exec_properties resolved by the launcher.

Returns
An DriverOutput instance.

SUPPORTED_EXECUTABLE_SPEC_TYPE