Finishing the renaming.

[flatland.git] / test.py

diff --git a/test.py b/test.py
index bf51360..314e03d 100755 (executable)
--- a/test.py
+++ b/test.py
@@ -40,10 +40,10 @@ def sequences_to_image(x):
 
 ######################################################################
 
-from _ext import mylib
+from _ext import flatland
 
 x = torch.ByteTensor()
 
-mylib.generate_sequence(10, x)
+flatland.generate_sequence(10, x)
 
 sequences_to_image(x).save('sequences.png')