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

tfx.v1.dsl.experimental.ResolverStrategy

Base resolver strategy class.

A resolver strategy defines a type behavior used for input selection. A resolver strategy subclass must override the resolve_artifacts() function which takes a dict of > as parameters and return the resolved dict.

Methods

resolve_artifacts

Resolves artifacts from channels, optionally querying MLMD if needed.

In asynchronous execution mode, resolver classes may composed in sequence where the resolve_artifacts() result from the previous resolver instance would be passed to the next resolver instance's resolve_artifacts() inputs.

If resolve_artifacts() returns None, it is considered as "no inputs available", and the remaining resolvers will not be executed.

Also if resolve_artifacts() omits any key from the input_dict it will not be available from the downstream resolver instances. General recommendation is to preserve all keys in the input_dict unless you have specific reason.

Args
store An MLMD MetadataStore.
input_dict The input_dict to resolve from.

Returns
If all entries has enough data after the resolving, returns the resolved input_dict. Otherise, return None.