X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?p=flatland.git;a=blobdiff_plain;f=test.py;h=c6b6c48c785a3b2a28a333c77f64af242906db79;hp=bf51360fbcf7a014071d3c0c64e746734767601d;hb=4aed0ce274b7c0e379651c28e439375c821c047a;hpb=c2a7c7d6dfec8bd1eca29406d160cce5b4a35209

diff --git a/test.py b/test.py
index bf51360..c6b6c48 100755
--- a/test.py
+++ b/test.py
@@ -4,6 +4,8 @@ import torch
 import torchvision
 from torchvision import datasets
 
+from _ext import flatland
+
 ######################################################################
 
 def sequences_to_image(x):
@@ -40,10 +42,6 @@ def sequences_to_image(x):
 
 ######################################################################
 
-from _ext import mylib
-
-x = torch.ByteTensor()
-
-mylib.generate_sequence(10, x)
+x = flatland.generate_sequence(5, 3, 128, 96)
 
 sequences_to_image(x).save('sequences.png')