X-Git-Url: https://fleuret.org/cgi-bin/gitweb/gitweb.cgi?a=blobdiff_plain;f=flatland.h;h=cdde100742f334fa1a4554c101b97d5124d3a4c0;hb=1e3df5a4bc6dfdf8fcc89270fbaf2cc43b82fef2;hp=e86533eae1cc2140369979645d43fb2d3240da68;hpb=1f91ec6f67da83525115f49dcc7d535ff2e71ef0;p=flatland.git

diff --git a/flatland.h b/flatland.h
index e86533e..cdde100 100644
--- a/flatland.h
+++ b/flatland.h
@@ -1,5 +1,7 @@
 
-THByteTensor *generate_sequence(long nb_sequences,
+THByteTensor *generate_sequence(int pulling,
+                                long nb_sequences,
                                 long nb_images,
                                 long image_height, long image_width,
+                                long nb_shapes,
                                 int random_shape_size, int random_colors);