From 6db695ad8a533b065884bd898de1c946bd31c9af Mon Sep 17 00:00:00 2001 From: shadiendrawis Date: Thu, 10 Oct 2019 17:47:25 +0300 Subject: [PATCH] freeze tensorflow version to <= 1.14.0 (#416) --- setup.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/setup.py b/setup.py index 512980f..850264c 100644 --- a/setup.py +++ b/setup.py @@ -73,11 +73,11 @@ if not using_GPU: # (reproduced with intel-tensorflow 1.14.0 but not with 1.13.1) install_requires.append('intel-tensorflow==1.13.1') else: - install_requires.append('tensorflow>=1.9.0') + install_requires.append('tensorflow>=1.9.0,<=1.14.0') extras['mxnet'] = ['mxnet-mkl>=1.3.0'] else: if not slim_package: - install_requires.append('tensorflow-gpu>=1.9.0') + install_requires.append('tensorflow-gpu>=1.9.0,<=1.14.0') extras['mxnet'] = ['mxnet-cu90mkl>=1.3.0'] all_deps = [] @@ -88,7 +88,7 @@ extras['all'] = all_deps setup( name='rl-coach' if not slim_package else 'rl-coach-slim', - version='1.0.0', + version='1.0.1', description='Reinforcement Learning Coach enables easy experimentation with state of the art Reinforcement Learning algorithms.', url='https://github.com/NervanaSystems/coach', author='Intel AI Lab',