feat(datasets) Add tests for mnist-m dataset (#3834)

adap · Jul 18, 2024 · f915835 · f915835
1 parent 4b94caa
commit f915835
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 0 deletions.
diff --git 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
@@ -34,6 +34,7 @@
     "svhn",
     "sentiment140",
     "speech_commands",
+    "Mike0307/MNIST-M",
     "flwrlabs/usps",
 ]