X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?p=flatland.git;a=blobdiff_plain;f=flatland-test.py;h=2c269ba0eb9dcc35dcb15ef7ca6df2bc9e0fddc3;hp=529284544d40b7bb23c008b09f6b7201e17f1bf6;hb=924ef5308c54784d2c0a9bbd282e55a16efbf318;hpb=d0756dc8a3e42aae30d1c7f543931a2316981635

diff --git a/flatland-test.py b/flatland-test.py
index 5292845..2c269ba 100755
--- 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
 
 ######################################################################
 
@@ -51,7 +51,7 @@ parser.add_argument('--nb_shapes',
                     help = 'Image height')
 
 parser.add_argument('--nb_sequences',
-                    type = int, default = 4,
+                    type = int, default = 8,
                     help = 'How many sequences to generate')
 
 parser.add_argument('--nb_images_per_sequences',