tfx.orchestration.portable.mlmd.context_lib.register_context_if_not_exists

Registers a context if not exist, otherwise returns the existing one.

tfx.orchestration.portable.mlmd.context_lib.register_context_if_not_exists(
    metadata_handler: tfx.orchestration.metadata.Metadata,
    context_type_name: Text,
    context_name: Text
) -> metadata_store_pb2.Context

This is a simplified wrapper around the method above which only takes context type and context name.

metadata_handler A handler to access MLMD store.
context_type_name The name of the context type.
context_name The name of the context.

An MLMD context.