diff options
Diffstat (limited to 'tensorflow/python/framework/versions_test.py')
| -rw-r--r-- | tensorflow/python/framework/versions_test.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/python/framework/versions_test.py b/tensorflow/python/framework/versions_test.py index b5bcc88e18..d94e892cd5 100644 --- a/tensorflow/python/framework/versions_test.py +++ b/tensorflow/python/framework/versions_test.py @@ -27,8 +27,8 @@ class VersionTest(tf.test.TestCase): self.assertEqual(type(tf.__version__), str) self.assertEqual(type(tf.VERSION), str) # This pattern will need to grow as we include alpha, builds, etc. - self.assertRegexpMatches(tf.__version__, r'^\d+\.\d+\.\w+$') - self.assertRegexpMatches(tf.VERSION, r'^\d+\.\d+\.\w+$') + self.assertRegexpMatches(tf.__version__, r'^\d+\.\d+\.(\d+(\-\w+)?|head)$') + self.assertRegexpMatches(tf.VERSION, r'^\d+\.\d+\.(\d+(\-\w+)?|head)$') def testGraphDefVersion(self): version = tf.GRAPH_DEF_VERSION |