Questions about TFX? Join us at Google I/O!

tfx.orchestration.kubeflow.v2.container.kubeflow_v2_entrypoint_utils.parse_raw_artifact_dict

Parses a map from key to a list of a single Artifact from pb objects.

tfx.orchestration.kubeflow.v2.container.kubeflow_v2_entrypoint_utils.parse_raw_artifact_dict(
    inputs_dict: Any,
    name_from_id: Optional[MutableMapping[int, str]] = None
) -> Dict[str, List[artifact.Artifact]]

Parses a mapping field in a protobuf message, whose value is an ExecutorInput.ArtifactList message, to a Python dict, whose value is a list of TFX Artifact Python objects.

inputs_dict the mapping field in the proto message.
name_from_id the dict used to store the id to string-typed name mapping.

dictionary of the parsed Python Artifact objects.