Replace `tensorflow.python.keras` with `keras`. `tensorflow.python.keras` is an old copy and is deprecated.
@@ -7,6 +7,7 @@ package(
default_visibility = [
":friends",
"//third_party/py/tensorflow:__subpackages__",
+ "//third_party/py/tensorflow_probability/python:__subpackages__",
"//third_party/tensorflow/python/feature_column:__subpackages__", # For unit test
"//third_party/tensorflow/python/tpu:__subpackages__", # For unit test
"//third_party/tensorflow_estimator:__subpackages__",
