Simplified naming and compilation.

[flatland.git] / flatland-test.py

diff --git a/flatland-test.py b/flatland-test.py
index 8592a4d..2c269ba 100755 (executable)
--- 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
 
 ######################################################################