fixing tensorflow dependency

kpe · Jun 5, 2019 · 468ea7c · 468ea7c
1 parent e58ed00
commit 468ea7c
Show file tree

Hide file tree

Showing 3 changed files with 3 additions and 2 deletions.
diff --git a/.travis.yml b/.travis.yml
@@ -17,7 +17,7 @@ env:
 # command to install dependencies
 install:
     - pip install --upgrade pip
-    - pip install tensorflow==2.0.0-alpha0
+    - pip install 'tensorflow>=1.14.0rc'
     - pip install -r requirements.txt
     - pip install coverage pep8 nose nose-exclude
 

diff --git a/requirements.txt b/requirements.txt
@@ -1,4 +1,5 @@
 
+tensorflow[gpu,tpu] >=1.14.0rc
 
 py-params   >= 0.6.3
 params-flow >= 0.2.2
diff --git a/version b/version
@@ -1 +1 @@
-0.1.6
+0.1.7