data indices creating changed to support uneven data arrays

Author: gaborvecsei

fed_learn/data_sampling.py

@@ -4,7 +4,7 @@
 def iid_data_indices(nb_clients: int, data_len: int):
     indices = np.arange(data_len)
     np.random.shuffle(indices)
-    chunks = np.split(indices, nb_clients)
+    chunks = np.array_split(indices, nb_clients)
     return chunks