Fixing import for mnist.
Fixing import for mnist.

PiperOrigin-RevId: 289527733

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