Simplified naming and compilation.

[flatland.git] / flatland-test.py

diff --git a/flatland-test.py b/flatland-test.py
index 8d9f11b..2c269ba 100755 (executable)
--- a/flatland-test.py
+++ b/flatland-test.py
@@ -25,7 +25,7 @@ import torch
 import torchvision
 import argparse
 
-from _ext import flatland
+import flatland
 
 ######################################################################