X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?p=flatland.git;a=blobdiff_plain;f=flatland.c;h=a77d7c1db77b0843e2bb74ca6a1393d6d9cd2578;hp=77f89aac193efe1c478dfecab1e8e03c1018cce2;hb=7e2780052c458a329fc36253b83eb5194ffb7fed;hpb=6c81a4b8ff7547e793d3f8bbd41ecdfdb73cbbbe

diff --git a/flatland.c b/flatland.c
index 77f89aa..a77d7c1 100644
--- a/flatland.c
+++ b/flatland.c
@@ -21,6 +21,7 @@
 THByteTensor *generate_sequence(long nb_sequences,
                                 long nb_images,
                                 long image_height, long image_width,
+                                int nb_shapes,
                                 int random_shape_size, int random_colors) {
 
   long nb_channels = 3;
@@ -46,7 +47,7 @@ THByteTensor *generate_sequence(long nb_sequences,
 
   for(s = 0; s < nb_sequences; s++) {
     a = THByteTensor_storage(result)->data + THByteTensor_storageOffset(result) + s * st0;
-    fl_generate_sequence(nb_images, image_width, image_height,
+    fl_generate_sequence(nb_images, image_width, image_height, nb_shapes,
                          random_shape_size, random_colors,
                          tmp_buffer);
     unsigned char *r = tmp_buffer;