Exclude pytorch in tests with python 3.8

geomstats · Jan 19, 2020 · 5959dde · 5959dde
1 parent 2894cf6
commit 5959dde
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/.travis.yml b/.travis.yml
@@ -14,7 +14,8 @@ install:
     - pip install -q -r requirements.txt --only-binary=numpy,scipy
 script:
     - flake8 examples geomstats tests
-    - if [[ $TRAVIS_PYTHON_VERSION != 3.8 || $GEOMSTATS_BACKEND != tensorflow ]]; then nose2 --with-coverage --verbose tests; fi
+    - if [[ $TRAVIS_PYTHON_VERSION != 3.8 ]]; then nose2 --with-coverage --verbose tests; fi
+    - if [[ $TRAVIS_PYTHON_VERSION == 3.8 && $GEOMSTATS_BACKEND == numpy ]]; then nose2 --with-coverage --verbose tests; fi
 env:
     - GEOMSTATS_BACKEND=numpy
     - GEOMSTATS_BACKEND=pytorch

diff --git a/requirements.txt b/requirements.txt
@@ -4,4 +4,4 @@ numpy>=1.18.1
 scikit-learn>=0.22.1
 scipy>=1.4.1
 tensorflow==2.0; python_version < '3.8'
-torch==1.3.1
+torch==1.3.1; python_version < '3.8'