X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?p=flatland.git;a=blobdiff_plain;f=build.py;h=307820907319245d5ee75aba26d5fbfc2bf07156;hp=4c3b766d04c000add1a5a54d87fcbddf15b805d8;hb=924ef5308c54784d2c0a9bbd282e55a16efbf318;hpb=c1f1040936d977cd2b3a276c725e223198377d2a

diff --git a/build.py b/build.py
index 4c3b766..3078209 100755
--- a/build.py
+++ b/build.py
@@ -1,4 +1,4 @@
-#!/usr/bin/env python-for-pytorch
+#!/usr/bin/env python
 
 #
 #  flatland is a simple 2d physical simulator
@@ -27,7 +27,7 @@ from torch.utils.ffi import create_extension
 abs_path = os.path.dirname(os.path.abspath(__file__))
 
 ffi = create_extension(
-    '_ext.flatland',
+    'flatland',
     headers = [ 'flatland.h' ],
     sources = [ 'flatland.c' ],
     extra_objects = [ abs_path + '/flatland_generator.so' ],