Also specify TF estimator

python-poetry/poetry#2271
pshved · Jul 8, 2020 · f8df0b6 · f8df0b6
1 parent 24f6acf
commit f8df0b6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/.travis.yml b/.travis.yml
@@ -19,8 +19,8 @@ install:
   - pip install poetry
   #- pip install tensorflow==$TF_VERSION
   # Work around https://github.com/python-poetry/poetry/issues/602
-  - poetry remove tensorflow
-  - poetry add tensorflow==$TF_VERSION
+  - poetry remove tensorflow tensorflow-estimator
+  - poetry add tensorflow==$TF_VERSION tensorflow-estimator==$TF_VERSION
   - poetry install -v
 
 script: pytest