Skip to content

Commit 0b76805

Browse files
Typo
1 parent 164d4fa commit 0b76805

File tree

1 file changed

+17
-9
lines changed

1 file changed

+17
-9
lines changed

setup.py

Lines changed: 17 additions & 9 deletions
Original file line numberDiff line numberDiff line change
@@ -10,12 +10,13 @@
1010
)
1111
except:
1212
raise ModuleNotFoundError("Please install pytorch >= 1.1 before proceeding.")
13-
13+
1414
import glob
1515

1616
from os import path
17+
1718
this_directory = path.abspath(path.dirname(__file__))
18-
with open(path.join(this_directory, 'README.md'), encoding='utf-8') as f:
19+
with open(path.join(this_directory, "README.md"), encoding="utf-8") as f:
1920
long_description = f.read()
2021

2122

@@ -27,7 +28,9 @@ def get_ext_modules():
2728
extra_compile_args += ["-DVERSION_GE_1_3"]
2829

2930
ext_src_root = "cuda"
30-
ext_sources = glob.glob("{}/src/*.cpp".format(ext_src_root)) + glob.glob("{}/src/*.cu".format(ext_src_root))
31+
ext_sources = glob.glob("{}/src/*.cpp".format(ext_src_root)) + glob.glob(
32+
"{}/src/*.cu".format(ext_src_root)
33+
)
3134

3235
ext_modules = []
3336
if CUDA_HOME:
@@ -36,7 +39,10 @@ def get_ext_modules():
3639
name="torch_points_kernels.points_cuda",
3740
sources=ext_sources,
3841
include_dirs=["{}/include".format(ext_src_root)],
39-
extra_compile_args={"cxx": extra_compile_args, "nvcc": extra_compile_args,},
42+
extra_compile_args={
43+
"cxx": extra_compile_args,
44+
"nvcc": extra_compile_args,
45+
},
4046
)
4147
)
4248

@@ -53,26 +59,28 @@ def get_ext_modules():
5359
)
5460
return ext_modules
5561

62+
5663
def get_cmdclass():
5764
return {"build_ext": BuildExtension}
5865

66+
5967
requirements = ["torch>=1.1.0"]
6068

61-
url = 'https://github.com/nicolas-chaulet/torch-points-kernels'
62-
__version__="0.6.3"
69+
url = "https://github.com/nicolas-chaulet/torch-points-kernels"
70+
__version__ = "0.6.3"
6371
setup(
6472
name="torch-points-kernels",
6573
version=__version__,
6674
author="Nicolas Chaulet",
6775
packages=find_packages(),
68-
description="PyTorch kernels for spatial operations on point clouds"
76+
description="PyTorch kernels for spatial operations on point clouds",
6977
url=url,
70-
download_url='{}/archive/{}.tar.gz'.format(url, __version__),
78+
download_url="{}/archive/{}.tar.gz".format(url, __version__),
7179
install_requires=requirements,
7280
ext_modules=get_ext_modules(),
7381
cmdclass=get_cmdclass(),
7482
long_description=long_description,
75-
long_description_content_type='text/markdown',
83+
long_description_content_type="text/markdown",
7684
classifiers=[
7785
"Programming Language :: Python :: 3",
7886
"License :: OSI Approved :: MIT License",

0 commit comments

Comments
 (0)