Remove one of the private import of tf.keras (#2655)
Remove one of the private import of tf.keras (#2655)

* Remove one of the private import of tf.keras

* Fix format issue

* More format fix

* Even more format fix

* Fix for flake8

* Remove the unnecessary white space.
tensorflow/addons,tensorflow/addons,tensorflow/addons