Merge pull request #143 from toslunar/test-chainerv3

Add CHAINER_VERSION config to CI
chainer · Sep 28, 2017 · 9c7136a · 9c7136a
2 parents 12cbb28 + 3a9a0a0
commit 9c7136a
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/.travis.yml b/.travis.yml
@@ -3,9 +3,18 @@ cache: pip
 python:
   - "2.7"
   - "3.5.1"
+env:
+  - CHAINER_VERSION=2
+  - CHAINER_VERSION=3
 # command to install dependencies
 install:
   - pip install --upgrade pip setuptools wheel
+  - |
+    if [ $CHAINER_VERSION -eq 2 ]; then
+      pip install "chainer<3.0.0"
+    else
+      pip install "chainer>=3.0.0rc1"
+    fi
   - pip install -r requirements.txt --only-binary=numpy,scipy
   - pip install jupyter
   - pip install atari_py>=0.1.1