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

diff --git a/test.py b/test.py
index 314e03d..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 flatland
-
-x = torch.ByteTensor()
-
-flatland.generate_sequence(10, x)
+x = flatland.generate_sequence(5, 3, 128, 96)
 
 sequences_to_image(x).save('sequences.png')