tfx.orchestration.portable.resolver_processor.ResolverStepProcessor

ResolverStepProcessor for processing single ResolverStep.

tfx.orchestration.portable.resolver_processor.ResolverStepProcessor(
    resolver_step: pipeline_pb2.ResolverConfig.ResolverStep
)

Note that input and the ouptut type of call is identical, thus resolver steps can be chained where the output of the former step would be fed into the next step. If the output is None, chained processing will be halted and the output of all steps would be considered None immediately.

Methods

__call__

View source

__call__(
    metadata_handler: tfx.orchestration.metadata.Metadata,
    input_dict: Mapping[str, Sequence[types.Artifact]]
) -> Optional[Dict[str, List[types.Artifact]]]

Resolves artifacts in input_dict by optionally querying MLMD.

Args
metadata_handler	A metadata handler to access MLMD store.
input_dict	Inputs to be resolved.

Returns
The resolved input_dict.