From: Mark Daoust <markdaoust@google.com>
Date: Mon, 4 Jun 2018 18:11:06 +0000 (-0700)
Subject: Fix visibility for tf.keras.__version__
X-Git-Tag: upstream/v1.9.0_rc1~8^2~3
X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=a08c8a79f3d0ea5a7fac74d8f5e9da5def89170b;p=platform%2Fupstream%2Ftensorflow.git

Fix visibility for tf.keras.__version__

PiperOrigin-RevId: 199161696
---

diff --git a/tensorflow/python/keras/__init__.py b/tensorflow/python/keras/__init__.py
index 197f306..3493069 100644
--- a/tensorflow/python/keras/__init__.py
+++ b/tensorflow/python/keras/__init__.py
@@ -41,8 +41,12 @@ from tensorflow.python.keras.layers import Input
 from tensorflow.python.keras.models import Model
 from tensorflow.python.keras.models import Sequential
 
+from tensorflow.python.util.tf_export import tf_export
+
 __version__ = '2.1.6-tf'
 
+tf_export('keras.__version__').export_constant(__name__, '__version__')
+
 del absolute_import
 del division
 del print_function
diff --git a/tensorflow/python/keras/integration_test.py b/tensorflow/python/keras/integration_test.py
index 2e83544..2a05699 100644
--- a/tensorflow/python/keras/integration_test.py
+++ b/tensorflow/python/keras/integration_test.py
@@ -29,6 +29,9 @@ from tensorflow.python.platform import test
 
 class KerasIntegrationTest(test.TestCase):
 
+  def test_version(self):
+    self.assertTrue(keras.__version__.endswith('-tf'))
+
   def test_vector_classification_sequential(self):
     with self.test_session():
       np.random.seed(1337)