Fixing bug in open source implementation of OGB-LSC/PCQ.
Fixing bug in open source implementation of OGB-LSC/PCQ.

PiperOrigin-RevId: 483319710

deepmind/deepmind-research,deepmind/deepmind-research,deepmind/deepmind-research,deepmind/deepmind-research