gaspersic/baselines
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Fix setup.py for tensorflow -rc versions

Browse Source
This commit is contained in:
Geoffrey Irving
2018-09-18 11:35:43 -07:00
parent 68a29d0ab3
commit a42c4eb2bb
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
setup.py
View File

@@ -1,3 +1,4 @@
import re
from setuptools import setup, find_packages from setuptools import setup, find_packages
import sys import sys
@@ -49,6 +50,6 @@ setup(name='baselines',
try: try:
from distutils.version import StrictVersion from distutils.version import StrictVersion
import tensorflow 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: except ImportError:
assert False, "TensorFlow needed, of version above 1.4" assert False, "TensorFlow needed, of version above 1.4"