baselines issue 146 (remove tensorflow from setup.py) (#34)

* baselines does not reinstall tensorflow

* fix the version check in baselines/setup.py

* replace print and assert with assert, str (thanks @csh)
This commit is contained in:
pzhokhov
2018-08-21 16:51:37 -07:00
committed by Peter Zhokhov
parent 20d22a5d79
commit 3900f2a447
2 changed files with 10 additions and 1 deletions

View File

@@ -33,7 +33,6 @@ setup(name='baselines',
'progressbar2',
'mpi4py',
'cloudpickle',
'tensorflow>=1.4.0',
'click',
'opencv-python'
],
@@ -43,3 +42,12 @@ setup(name='baselines',
url='https://github.com/openai/baselines',
author_email='gym@openai.com',
version='0.1.5')
# ensure there is some tensorflow build with version above 1.4
try:
from distutils.version import StrictVersion
import tensorflow
assert StrictVersion(tensorflow.__version__) >= StrictVersion('1.4.0')
except ImportError:
assert False, "TensorFlow needed, of version above 1.4"