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,
    tfx_ir: Optional[pipeline_pb2.Pipeline] = None,
    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.
`tfx_ir`	The TFX intermedia representation of the pipeline.
`pod_labels_to_attach`	Optional dict of pod labels to attach to the GKE pod.

tfx.orchestration.kubeflow.base_component.BaseComponent

Args