Replace `tensorflow.python.keras` with `keras`. `tensorflow.python.keras` is an old copy and is deprecated.
PiperOrigin-RevId: 485920027
A
A. Unique TensorFlower committed
6cca0b6a5069e97cb952d947cf2e2af8efeaf8ab
Parent: c2c88d4