From 7ccea71ef9bc1204ac42a0cb57f717fec43e8a24 Mon Sep 17 00:00:00 2001
From: Stephan Seitz <stephan.seitz@fau.de>
Date: Fri, 13 Mar 2020 22:02:42 +0100
Subject: [PATCH] Remove some unnecessary statements in setup.py
---
setup.py | 4 ++--
1 file changed, 2 insertions(+), 2 deletions(-)
diff --git a/setup.py b/setup.py
index e774a6c..b85a04d 100644
--- a/setup.py
+++ b/setup.py
@@ -1,9 +1,10 @@
import sys
from glob import glob
-from os.path import dirname, join
+from os.path import join
from pkg_resources import VersionConflict, require
from setuptools import setup
+
from torch.utils.cpp_extension import BuildExtension, CUDAExtension
try:
@@ -15,7 +16,6 @@ except VersionConflict:
if __name__ == "__main__":
- object_cache = dirname(__file__)
module_name = 'pyronn_torch_cpp'
cuda_sources = glob(join('generated_files', '*.cu'))
--
GitLab