tfx.orchestration.kubeflow.base_component.BaseComponent

Base component for all Kubeflow pipelines TFX components.

tfx.orchestration.kubeflow.base_component.BaseComponent(
    component: tfx.dsl.components.base.base_node.BaseNode,
    component_launcher_class: Type[tfx.orchestration.launcher.base_component_launcher.BaseComponentLauncher],
    depends_on: Set[dsl.ContainerOp],
    pipeline: tfx.orchestration.pipeline.Pipeline,
    pipeline_name: Text,
    pipeline_root: dsl.PipelineParam,
    tfx_image: Text,
    kubeflow_metadata_config: Optional[tfx.orchestration.kubeflow.proto.kubeflow_pb2.KubeflowMetadataConfig],
    component_config: tfx.orchestration.config.base_component_config.BaseComponentConfig,
    pod_labels_to_attach: Optional[Dict[Text, Text]] = None
)

Returns a wrapper around a KFP DSL ContainerOp class, and adds named output attributes that match the output names for the corresponding native TFX components.

Args
component	The logical TFX component to wrap.
component_launcher_class	the class of the launcher to launch the component.
depends_on	The set of upstream KFP ContainerOp components that this component will depend on.
pipeline	The logical TFX pipeline to which this component belongs.
pipeline_name	The name of the TFX pipeline.
pipeline_root	The pipeline root specified, as a dsl.PipelineParam
tfx_image	The container image to use for this component.
kubeflow_metadata_config	Configuration settings for connecting to the MLMD store in a Kubeflow cluster.
component_config	Component config to launch the component.
pod_labels_to_attach	Optional dict of pod labels to attach to the GKE pod.