tfx.dsl.components.base.base_executor.BaseExecutor

View source on GitHub

Abstract TFX executor class.

tfx.dsl.components.base.base_executor.BaseExecutor(
    context: Optional[tfx.dsl.components.base.base_executor.BaseExecutor.Context] = None
)

Child Classes

class Context

Methods

Do

View source

@abc.abstractmethod
Do(
    input_dict: Dict[Text, List[tfx.types.Artifact]],
    output_dict: Dict[Text, List[tfx.types.Artifact]],
    exec_properties: Dict[Text, Any]
) -> Optional[execution_result_pb2.ExecutorOutput]

Execute underlying component implementation.

Args
input_dict	Input dict from input key to a list of Artifacts. These are often outputs of another component in the pipeline and passed to the component by the orchestration system.
output_dict	Output dict from output key to a list of Artifacts. These are often consumed by a dependent component.
exec_properties	A dict of execution properties. These are inputs to pipeline with primitive types (int, string, float) and fully materialized when a pipeline is constructed. No dependency to other component or later injection from orchestration systems is necessary or possible on these values.

Returns
execution_result_pb2.ExecutorOutput or None.