1 files changed, 4 insertions, 1 deletions
diff --git a/km-init-rand.c b/km-init-rand.c
index f9d6eb6..c6000bd 100644
--- a/km-init-rand.c
+++ b/km-init-rand.c
@@ -9,9 +9,12 @@ bool
 km_init_rand(
   km_set_t * const cs,
   const size_t num_clusters,
-  const size_t num_floats,
+  const km_set_t * const set,
   km_rand_t * const rs
 ) {
+  // get number of floats from data set
+  const size_t num_floats = set->shape.num_floats;
+
   // init cluster shape
   const km_shape_t shape = {
     .num_floats = num_floats,