Join the SIG TFX-Addons community and help make TFX even better!

tfx.orchestration.publisher.Publisher

Publish execution to metadata.

tfx.orchestration.publisher.Publisher(
    metadata_handler: tfx.orchestration.metadata.Metadata
)

_metadata_handler An instance of Metadata.

Methods

publish_execution

View source

publish_execution(
    component_info: tfx.orchestration.data_types.ComponentInfo,
    output_artifacts: Optional[Dict[Text, List[types.Artifact]]] = None,
    exec_properties: Optional[Dict[Text, Any]] = None
)

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
component_info the information of the component
output_artifacts optional key -> Artifacts to be published as outputs of the execution
exec_properties optional execution properties to be published for the execution

Returns
A dict containing output artifacts.