set numpy version <1.20

Author: clee-ai

.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 numba
+                  pip install torch "numpy<1.20" scikit-learn flake8 setuptools wheel twine numba
             - name: Build package
               run: |
                   python setup.py build_ext --inplace

setup.py

@@ -21,8 +21,8 @@
     WITH_CUDA = True
     WITH_CPU = False
 if os.getenv('FORCE_ONLY_CPU', '0') == '1':
-    WITH_CPU = True
     WITH_CUDA = False
+    WITH_CPU = True
 
 def get_ext_modules():
     TORCH_MAJOR = int(torch.__version__.split(".")[0])
@@ -77,7 +77,7 @@ def get_cmdclass():
 with open(os.path.join(this_directory, "README.md"), encoding="utf-8") as f:
     long_description = f.read()
 
-requirements = ["torch>=1.1.0", "numba", "scikit-learn"]
+requirements = ["torch>=1.1.0", "numba", "numpy<1.20", "scikit-learn"]
 
 url = "https://github.com/nicolas-chaulet/torch-points-kernels"
 __version__ = "0.6.10"