tfma.metrics.merge_per_key_computations

Wraps create_computations_fn to be called separately for each key.