Update.

[flatland.git] / flatland.h

diff --git a/flatland.h b/flatland.h
index e86533e..ba91f67 100644 (file)
--- 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);