tfx.orchestration.publisher.Publisher

View source on GitHub

Class Publisher

Publish execution to metadata.

Attributes:

• _metadata_handler: An instance of Metadata.

__init__

View source

__init__(metadata_handler)

Initialize self. See help(type(self)) for accurate signature.

Methods

publish_execution

View source

publish_execution(
    execution_id,
    input_dict,
    output_dict,
    use_cached_results
)

Publishes a component execution to metadata.

This function will do two things:

1. update the execution that was previously registered before execution to complete or skipped state, depending on whether cached results are used.
2. for each input and output artifact, publish an event that associate the artifact to the execution, with type INPUT or OUTPUT respectively

Args:

• execution_id: the execution id for the
• input_dict: key -> Artifacts that are used as inputs in the execution
• output_dict: key -> Artifacts that are declared as outputs for the execution
• use_cached_results: whether or not the execution has used cached results

Returns:

A dict containing output artifacts.