message: Fix build failure for mac/ubuntu, which relies on an old version for keras-preprocessing.
diff: @@ -18,6 +18,17 @@ from __future__ import absolute_import
  from __future__ import division
  from __future__ import print_function
  +# TODO(mihaimaruseac): remove the import of keras_preprocessing and injecting
  +# once we update to latest version of keras_preprocessing
  +import keras_preprocessing
  +
  +from tensorflow.python.keras import backend
  +from tensorflow.python.keras.utils import all_utils as utils
  +
  +# This exists for compatibility with prior version of keras_preprocessing.
  +keras_preprocessing.set_keras_submodules(backend=backend, utils=utils)
  +
  +
  from tensorflow.python.keras.preprocessing import image
  from tensorflow.python.keras.preprocessing import sequence
  from tensorflow.python.keras.preprocessing import text
  