![]() |
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. |