Skip to content
Snippets Groups Projects
Commit 7ccea71e authored by Stephan Seitz's avatar Stephan Seitz
Browse files

Remove some unnecessary statements in setup.py

parent 102157fd
No related merge requests found
import sys import sys
from glob import glob from glob import glob
from os.path import dirname, join from os.path import join
from pkg_resources import VersionConflict, require from pkg_resources import VersionConflict, require
from setuptools import setup from setuptools import setup
from torch.utils.cpp_extension import BuildExtension, CUDAExtension from torch.utils.cpp_extension import BuildExtension, CUDAExtension
try: try:
...@@ -15,7 +16,6 @@ except VersionConflict: ...@@ -15,7 +16,6 @@ except VersionConflict:
if __name__ == "__main__": if __name__ == "__main__":
object_cache = dirname(__file__)
module_name = 'pyronn_torch_cpp' module_name = 'pyronn_torch_cpp'
cuda_sources = glob(join('generated_files', '*.cu')) cuda_sources = glob(join('generated_files', '*.cu'))
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment