increment pytorch versions for testing (#133)

* imcrement pytorch versions for testing * cpu only tests
geoopt · May 23, 2020 · 67f5a9e · 67f5a9e
1 parent 1f067dc
commit 67f5a9e
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 4 deletions.
diff --git a/.travis.yml b/.travis.yml
@@ -9,10 +9,10 @@
 language : shell
 
 env:
-  - PYTHON_VERSION=3.6 PYTORCH_CHANNEL='pytorch' PYTORCH_VERSION='=1.4.0' COVERAGE='--cov geoopt'
+  - PYTHON_VERSION=3.6 PYTORCH_CHANNEL='pytorch' PYTORCH_VERSION='=1.5.0' COVERAGE='--cov geoopt'
+  - PYTHON_VERSION=3.7 PYTORCH_CHANNEL='pytorch' PYTORCH_VERSION='=1.5.0' COVERAGE=''
+  - PYTHON_VERSION=3.6 PYTORCH_CHANNEL='pytorch' PYTORCH_VERSION='=1.4.0' COVERAGE=''
   - PYTHON_VERSION=3.7 PYTORCH_CHANNEL='pytorch' PYTORCH_VERSION='=1.4.0' COVERAGE=''
-  - PYTHON_VERSION=3.6 PYTORCH_CHANNEL='pytorch' PYTORCH_VERSION='=1.3.1' COVERAGE=''
-  - PYTHON_VERSION=3.7 PYTORCH_CHANNEL='pytorch' PYTORCH_VERSION='=1.3.1' COVERAGE=''
   - PYTHON_VERSION=3.7 PYTORCH_CHANNEL='pytorch-nightly' PYTORCH_VERSION='' COVERAGE=''
 
 matrix:

diff --git a/CHANGELOG.rst b/CHANGELOG.rst
@@ -7,6 +7,9 @@ New Features
 ------------
 * BirkhoffPolytope (#125)
 
+Maintenance
+-----------
+* Tests for pytorch>=1.4, cpuonly (#133)
 
 geoopt v0.1.2
 ==============

diff --git a/scripts/install_deps.sh b/scripts/install_deps.sh
@@ -1,3 +1,3 @@
 conda install --yes pip numpy mkl-service
-conda install --yes pytorch${PYTORCH_VERSION} -c ${PYTORCH_CHANNEL}
+conda install --yes pytorch${PYTORCH_VERSION} cpuonly -c ${PYTORCH_CHANNEL}
 pip install -r ./requirements-dev.txt