Specify a tensorflow version, since the code is not tf-2 compatible.
Specify a tensorflow version, since the code is not tf-2 compatible.

PiperOrigin-RevId: 324897075

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