Pin pytorch version

Author: nicolas-chaulet

.github/workflows/tests.yaml

@@ -22,7 +22,8 @@ jobs:
         - name: Install dependencies
           run: |
               python -m pip install --upgrade pip
-              pip install torch numpy scikit-learn flake8 setuptools
+              pip install numpy scikit-learn flake8 setuptools
+              pip install torch==1.5.0+cpu torchvision==0.6.0+cpu -f https://download.pytorch.org/whl/torch_stable.html
         - name: Build package
           run: |
               python setup.py build_ext --inplace