Handle missing torch better

Author: nicolas-chaulet

CHANGELOG.md

@@ -1,10 +1,15 @@
 # Unreleased
 
+# 0.6.0
+
+## Bug fix
+- Require pytorch implicitely and log nice message when missing
+
 # 0.5.3
 
 ## Update
 - ball query returns squared distance instead of distance
 - leaner Point Cloud struct that avoids copying data
 
 ## Bug fix
-- Pcakage would not install if pytorch is not already installed
+- Package would not install if pytorch is not already installed

pyproject.toml

@@ -25,5 +25,5 @@ exclude = '''
 '''
 
 [build-system]
-requires = ["setuptools>=41.0", "setuptools-scm", "wheel", "torch"]
+requires = ["setuptools>=41.0", "setuptools-scm", "wheel"]
 build-backend = "setuptools.build_meta"

setup.py

@@ -1,5 +1,10 @@
 from setuptools import setup, find_packages
-import torch
+
+try:
+    import torch
+except:
+    raise ImportError("Please install pytorch before installing torch-points-kernels")
+
 from torch.utils.cpp_extension import (
     BuildExtension,
     CUDAExtension,
@@ -48,7 +53,7 @@
 requirements = ["torch>=1.1.0"]
 
 url = 'https://github.com/nicolas-chaulet/torch-points-kernels'
-__version__="0.5.3"
+__version__="0.6.0"
 setup(
     name="torch-points-kernels",
     version=__version__,