From f91583541808aec206d2f6d7ffbccf749d6b2245 Mon Sep 17 00:00:00 2001
From: Adam Narozniak <51029327+adam-narozniak@users.noreply.github.com>
Date: Thu, 18 Jul 2024 18:10:22 +0200
Subject: [PATCH] feat(datasets) Add tests for `mnist-m` dataset (#3834)

---
 datasets/flwr_datasets/federated_dataset_test.py | 1 +
 datasets/flwr_datasets/utils.py                  | 1 +
 2 files changed, 2 insertions(+)

diff --git a/datasets/flwr_datasets/federated_dataset_test.py b/datasets/flwr_datasets/federated_dataset_test.py
index 9faf5d93afc9..2f6bdbd6376d 100644
--- a/datasets/flwr_datasets/federated_dataset_test.py
+++ b/datasets/flwr_datasets/federated_dataset_test.py
@@ -43,6 +43,7 @@
         ("fashion_mnist", "test", ""),
         ("sasha/dog-food", "test", ""),
         ("zh-plus/tiny-imagenet", "valid", ""),
+        ("Mike0307/MNIST-M", "test", ""),
         ("flwrlabs/usps", "test", ""),
         # Text
         ("scikit-learn/adult-census-income", None, ""),
diff --git a/datasets/flwr_datasets/utils.py b/datasets/flwr_datasets/utils.py
index 2c58340233d9..00d1f10756ae 100644
--- a/datasets/flwr_datasets/utils.py
+++ b/datasets/flwr_datasets/utils.py
@@ -34,6 +34,7 @@
     "svhn",
     "sentiment140",
     "speech_commands",
+    "Mike0307/MNIST-M",
     "flwrlabs/usps",
 ]