Merge branch 'main' of github.com:QinbinLi/MOON into main

WangZM333 · Apr 7, 2021 · dbf6344 · dbf6344
2 parents fa0c35a + cd8982f
commit dbf6344
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/utils.py b/utils.py
@@ -98,7 +98,7 @@ def partition_data(dataset, datadir, logdir, partition, n_parties, beta=0.4):
 
     n_train = y_train.shape[0]
 
-    elif partition == "homo" or partition == "iid":
+    if partition == "homo" or partition == "iid":
         idxs = np.random.permutation(n_train)
         batch_idxs = np.array_split(idxs, n_parties)
         net_dataidx_map = {i: batch_idxs[i] for i in range(n_parties)}