Numba 0.49.1

Author: nicolas-chaulet

.github/workflows/tests.yaml

@@ -22,15 +22,15 @@ jobs:
         - name: Install dependencies
           run: |
               python -m pip install --upgrade pip
-              pip install numpy scikit-learn flake8 setuptools numba
-              
+              pip install numpy scikit-learn flake8 setuptools numba==0.49.1
+
         - name: Install torch windows + linux
           if: ${{matrix.os != 'macos-latest'}}
           run: pip install torch==1.5.0+cpu torchvision==0.6.0+cpu -f https://download.pytorch.org/whl/torch_stable.html
         - name: Install torch macos
           if: ${{matrix.os == 'macos-latest'}}
-          run: pip install torch 
-            
+          run: pip install torch
+
         - name: Build package
           run: |
               python setup.py build_ext --inplace