Update utils.py

Xtra-Computing · Nov 17, 2021 · 42e6035 · 42e6035
1 parent 3af760c
commit 42e6035
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/utils.py b/utils.py
@@ -360,7 +360,7 @@ def partition_data(dataset, datadir, logdir, partition, n_parties, beta=0.4):
         batch_idxs = np.split(idxs,proportions)
         net_dataidx_map = {i: batch_idxs[i] for i in range(n_parties)}
 
-elif partition == "mixed":
+    elif partition == "mixed":
         min_size = 0
         min_require_size = 10
         K = 10