X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?p=flatland.git;a=blobdiff_plain;f=flatland-test.py;h=2c269ba0eb9dcc35dcb15ef7ca6df2bc9e0fddc3;hp=8592a4db3fe57cd08ff99eb287dc923b2fc5a98e;hb=924ef5308c54784d2c0a9bbd282e55a16efbf318;hpb=b8076332f83b24a86e727b4e4399b5965a983e7a

diff --git a/flatland-test.py b/flatland-test.py
index 8592a4d..2c269ba 100755
--- a/flatland-test.py
+++ b/flatland-test.py
@@ -1,4 +1,4 @@
-#!/usr/bin/env python-for-pytorch
+#!/usr/bin/env python
 
 #
 #  flatland is a simple 2d physical simulator
@@ -25,7 +25,7 @@ import torch
 import torchvision
 import argparse
 
-from _ext import flatland
+import flatland
 
 ######################################################################