Merge pull request #24 from nicolas-chaulet/renaming

Renaming

Author: nicolas-chaulet

.github/workflows/deploy.yaml

@@ -0,0 +1,40 @@
+on:
+  push:
+    tags:
+    - 'v*'
+    
+name: Deploy
+
+jobs:
+    deploy:
+        runs-on: ubuntu-latest
+        steps:
+            - uses: actions/checkout@v1
+            - name: Set up Python 3.6
+              uses: actions/setup-python@v1
+              with:
+                  python-version: 3.6
+            - name: Install dependencies
+              run: |
+                  python -m pip install --upgrade pip
+                  pip install torch numpy scikit-learn flake8 setuptools wheel twine
+            - name: Build package
+              run: |
+                  python setup.py build_ext --inplace
+            - name: Lint with flake8
+              run: |
+                  # stop the build if there are Python syntax errors or undefined names
+                  flake8 . --count --select=E9,F63,F7,F82 --show-source --statistics
+                  # exit-zero treats all errors as warnings. The GitHub editor is 127 chars wide
+                  flake8 . --count --exit-zero --max-complexity=10 --max-line-length=127 --statistics
+            - name: Test with unittest
+              run: |
+                  python -m unittest -v
+            - name: Build package
+              run: |
+                  python setup.py sdist
+            - name: Publish package
+              uses: pypa/gh-action-pypi-publish@master
+              with:
+                user: __token__
+                password: ${{ secrets.PYPI_PASSWORD }}

.gitignore

@@ -36,4 +36,4 @@ build
 
 .vscode/
 dist/
-torch_points.egg-info/
+torch_points_kernels.egg-info/

README.md

@@ -1,20 +1,20 @@
-# torch-points
-Pytorch kernels for pointnet++ like architectures
+# 3D Point Cloud Kernels
+Pytorch CPU and CUDA kernels for spatial search and interpolation for 3D point clouds.
 
 ## Installation
 Requires torch version 1.0 or higher. To install a specific version replace the version number with a given tag.
 ```
-pip install git+https://github.com/nicolas-chaulet/torch-points.git#v0.2.3
+pip install git+https://github.com/nicolas-chaulet/torch-points-kernels.git#v0.5.0
 ```
 or with poetry:
 ```
-poetry add git+https://github.com/nicolas-chaulet/torch-points.git#v0.2.3
+poetry add git+https://github.com/nicolas-chaulet/torch-points.git#v0.5.0
 ```
 
 ## Usage
 ```
 import torch
-import torch_points.points_cuda
+import torch_points_kernels.points_cuda
 ```
 
 ## Build and test
@@ -25,10 +25,7 @@ python -m unittest
 
 ## Projects using those kernels.
 
-[```Pytorch Point Cloud Benchmark```](https://github.com/nicolas-chaulet/deeppointcloud-benchmarks) by
-* [Thomas Chaton](https://github.com/tchaton)
-* Nicolas Chaulet
-* [Tristan Heywood](https://github.com/tristanheywood)
+[```Pytorch Point Cloud Benchmark```](https://github.com/nicolas-chaulet/deeppointcloud-benchmarks)
 
 ## Credit
 

setup.py

@@ -8,6 +8,11 @@
 )
 import glob
 
+from os import path
+this_directory = path.abspath(path.dirname(__file__))
+with open(path.join(this_directory, 'README.md'), encoding='utf-8') as f:
+    long_description = f.read()
+
 TORCH_MAJOR = int(torch.__version__.split(".")[0])
 TORCH_MINOR = int(torch.__version__.split(".")[1])
 extra_compile_args = []
@@ -21,7 +26,7 @@
 if CUDA_HOME:
     ext_modules.append(
         CUDAExtension(
-            name="torch_points.points_cuda",
+            name="torch_points_kernels.points_cuda",
             sources=ext_sources,
             include_dirs=["{}/include".format(ext_src_root)],
             extra_compile_args={"cxx": extra_compile_args, "nvcc": extra_compile_args,},
@@ -33,7 +38,7 @@
 
 ext_modules.append(
     CppExtension(
-        name="torch_points.points_cpu",
+        name="torch_points_kernels.points_cpu",
         sources=cpu_ext_sources,
         include_dirs=["{}/include".format(cpu_ext_src_root)],
         extra_compile_args={"cxx": extra_compile_args,},
@@ -42,12 +47,18 @@
 
 requirements = ["torch>=1.1.0"]
 
+url = 'https://github.com/nicolas-chaulet/torch-points-kernels'
+__version__="0.5.1"
 setup(
-    name="torch_points",
-    version="0.4.1",
+    name="torch-points-kernels",
+    version=__version__,
     author="Nicolas Chaulet",
     packages=find_packages(),
+    url=url,
+    download_url='{}/archive/{}.tar.gz'.format(url, __version__),
     install_requires=requirements,
     ext_modules=ext_modules,
     cmdclass={"build_ext": BuildExtension},
+    long_description=long_description,
+    long_description_content_type='text/markdown'
 )

test/test_ballquerry.py

@@ -1,6 +1,6 @@
 import unittest
 import torch
-from torch_points import ball_query
+from torch_points_kernels import ball_query
 import numpy.testing as npt
 import numpy as np
 from sklearn.neighbors import KDTree

test/test_fps.py

@@ -6,7 +6,7 @@
 ROOT = os.path.join(os.path.dirname(os.path.realpath(__file__)), "..")
 sys.path.insert(0, ROOT)
 
-from torch_points.points_cpu import fps
+from torch_points_kernels.points_cpu import fps
 
 
 class TestFps(unittest.TestCase):

test/test_grouping.py

@@ -2,7 +2,7 @@
 import torch
 import numpy as np
 import numpy.testing as npt
-from torch_points import grouping_operation
+from torch_points_kernels import grouping_operation
 
 
 class TestGroup(unittest.TestCase):

test/test_interpolate.py

@@ -1,7 +1,7 @@
 import unittest
 import torch
 from torch.autograd import gradcheck
-from torch_points import three_interpolate, three_nn
+from torch_points_kernels import three_interpolate, three_nn
 
 from . import run_if_cuda
 

test/test_knn.py

@@ -6,7 +6,7 @@
 ROOT = os.path.join(os.path.dirname(os.path.realpath(__file__)), "..")
 sys.path.insert(0, ROOT)
 
-from torch_points import three_nn, knn
+from torch_points_kernels import three_nn, knn
 from . import run_if_cuda