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)

Dockerfile

@@ -18,6 +18,7 @@ WORKDIR $CODE_DIR/baselines
 # Clean up pycache and pyc files
 RUN rm -rf __pycache__ && \
     find . -name "*.pyc" -delete && \
+    pip install tensorflow && \
     pip install -e .[test]
 
 

setup.py

@@ -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"