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

tfx.orchestration.portable.mlmd.execution_lib.get_artifact_ids_by_event_type_for_execution_id

Returns artifact ids corresponding to the execution id grouped by event type.

tfx.orchestration.portable.mlmd.execution_lib.get_artifact_ids_by_event_type_for_execution_id(
    metadata_handler: tfx.orchestration.metadata.Metadata,
    execution_id: int
) -> Dict['metadata_store_pb2.Event.Type', Set[int]]

metadata_handler A handler to access MLMD.
execution_id Id of the execution for which to get artifact ids.

A dict mapping event type to set of artifact ids.