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

diff --git a/flatland.h b/flatland.h
index e86533e..ba91f67 100644
--- a/flatland.h
+++ b/flatland.h
@@ -2,4 +2,5 @@
 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);