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

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
 
 ######################################################################