Make this code compatible with Python 3.10.
Make this code compatible with Python 3.10.

PiperOrigin-RevId: 473337535

google-research/google-research,google-research/google-research,google-research/google-research,google-research/google-research,google-research/google-research,google-research/google-research,google-research/google-research,google-research/google-research,google-research/google-research,google-research/google-research,google-research/google-research