X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?p=pysvrt.git;a=blobdiff_plain;f=build.py;h=1a10c62396c3314ab4a6a74206811e27dd585204;hp=ef0b46d24b3fd05bc3ca74bbb334e6ad7027115e;hb=15f2d2cf0a655234cfa435789e26238b95f5a371;hpb=f542d0542b1e51ca7dd12bc6b96f6a299371ae8d

diff --git a/build.py b/build.py
index ef0b46d..1a10c62 100755
--- a/build.py
+++ b/build.py
@@ -21,21 +21,17 @@
 #  You should have received a copy of the GNU General Public License
 #  along with selector.  If not, see <http://www.gnu.org/licenses/>.
 
-import os
+from os import path
 from torch.utils.ffi import create_extension
 
-abs_path = os.path.dirname(os.path.abspath(__file__))
+abs_path = path.dirname(path.abspath(__file__))
 
 ffi = create_extension(
-    '_ext.svrt',
+    'svrt',
     headers = [ 'svrt.h' ],
     sources = [ 'svrt.c' ],
     extra_objects = [ abs_path + '/libsvrt.so' ],
-    libraries = [ ],
-    library_dirs = [ ],
-    define_macros = [ ],
     with_cuda = False
 )
 
-if __name__ == '__main__':
-    ffi.build()
+ffi.build()