Adding numba

Author: nicolas-chaulet

.devcontainer/Dockerfile

@@ -29,5 +29,5 @@ RUN apt-get update \
     && rm -rf /var/lib/apt/lists/*
 
 RUN pip3 install -U pip
-RUN pip3 install torch numpy scikit-learn flake8 setuptools
+RUN pip3 install torch numpy scikit-learn flake8 setuptools numba
 RUN pip3 install torch_cluster torch_sparse torch_scatter torch_geometric

.github/workflows/deploy.yaml

@@ -17,7 +17,7 @@ jobs:
             - name: Install dependencies
               run: |
                   python -m pip install --upgrade pip
-                  pip install torch numpy scikit-learn flake8 setuptools wheel twine
+                  pip install torch numpy scikit-learn flake8 setuptools wheel twine numba
             - name: Build package
               run: |
                   python setup.py build_ext --inplace

.github/workflows/tests.yaml

@@ -22,14 +22,14 @@ jobs:
         - name: Install dependencies
           run: |
               python -m pip install --upgrade pip
-              pip install numpy scikit-learn flake8 setuptools
+              pip install numpy scikit-learn flake8 setuptools numba
               
         - 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: |

setup.py

@@ -64,7 +64,7 @@ def get_cmdclass():
     return {"build_ext": BuildExtension}
 
 
-requirements = ["torch>=1.1.0"]
+requirements = ["torch>=1.1.0", "numba"]
 
 url = "https://github.com/nicolas-chaulet/torch-points-kernels"
 __version__ = "0.6.4"