From d7e1e5755a6444159edbfb91aab62f688132da5a Mon Sep 17 00:00:00 2001
From: rishi-ps <127986510+rishi-ps@users.noreply.github.com>
Date: Sun, 21 Sep 2025 01:23:08 +0530
Subject: [PATCH] fix typo in comment about number of shards

---
 src/sampling.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/sampling.py b/src/sampling.py
index ee0a5f2c..e3121e33 100755
--- a/src/sampling.py
+++ b/src/sampling.py
@@ -30,7 +30,7 @@ def mnist_noniid(dataset, num_users):
     :param num_users:
     :return:
     """
-    # 60,000 training imgs -->  200 imgs/shard X 300 shards
+    # 60,000 training imgs -->  300 imgs/shard X 200 shards
     num_shards, num_imgs = 200, 300
     idx_shard = [i for i in range(num_shards)]
     dict_users = {i: np.array([]) for i in range(num_users)}