tfx.orchestration.component_launcher.ComponentLauncher

View source on GitHub

Class ComponentLauncher

Responsible for launching driver, executor and publisher of component.

Inherits From: BaseComponentLauncher

Convenient subclass when given a concrete component to launch.

__init__

View source

__init__(
    component,
    pipeline_info,
    driver_args,
    metadata_connection_config,
    additional_pipeline_args
)

Initialize a ComponentLauncher.

Args:

  • component: The component to launch.
  • pipeline_info: An instance of data_types.PipelineInfo that holds pipeline properties.
  • driver_args: An instance of data_types.DriverArgs that holds component specific driver args.
  • metadata_connection_config: ML metadata connection config.
  • additional_pipeline_args: Additional pipeline args, includes,
    • beam_pipeline_args: Beam pipeline args for beam jobs within executor. Executor will use beam DirectRunner as Default.

Methods

launch

View source

launch()

Execute the component, includes driver, executor and publisher.

Returns:

The execution id of the launch.