Fix setup.py for tensorflow -rc versions

2018-09-18 11:35:43 -07:00
parent 68a29d0ab3
commit a42c4eb2bb
1 changed files with 2 additions and 1 deletions
--- a/setup.py
+++ b/setup.py
@@ -1,3 +1,4 @@
+import re
 from setuptools import setup, find_packages
 import sys

@@ -49,6 +50,6 @@ setup(name='baselines',
 try:
    from distutils.version import StrictVersion
    import tensorflow
-    assert StrictVersion(tensorflow.__version__) >= StrictVersion('1.4.0')
+    assert StrictVersion(re.sub(r'-rc\d+$', '', tensorflow.__version__)) >= StrictVersion('1.4.0')
 except ImportError:
    assert False, "TensorFlow needed, of version above 1.4"