Minor fixes + changed the default number of epochs to 100.

[pysvrt.git] / svrt_generator.h

diff --git a/svrt_generator.h b/svrt_generator.h
index bdfe5c1..7f6a3ad 100644 (file)
--- a/svrt_generator.h
+++ b/svrt_generator.h
@@ -35,7 +35,8 @@ struct VignetteSet {
   unsigned char *data;
 };
 
-void svrt_generate_vignettes(int n_problem, int nb_vignettes, struct VignetteSet *result);
+  void svrt_generate_vignettes(int n_problem, int nb_vignettes, long *labels,
+                               struct VignetteSet *result);
 
 #ifdef __cplusplus
 }