-#!/usr/bin/env python-for-pytorch
+#!/usr/bin/env python
#
# flatland is a simple 2d physical simulator
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' ],