Adds `_uncertain_retrieval_on_gold` to support getting retrieval examplars based on the corresponding gold subgraphs regarding uncertain parts of the prediction.
Adds `_uncertain_retrieval_on_gold` to support getting retrieval examplars based on the corresponding gold subgraphs regarding uncertain parts of the prediction.

PiperOrigin-RevId: 466701635

google/uncertainty-baselines