feat(datasets) Add tests for uci-mushrooms dataset (#3841)

adap · Jul 18, 2024 · ddb1eec · ddb1eec
1 parent f915835
commit ddb1eec
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
@@ -47,6 +47,7 @@
         ("flwrlabs/usps", "test", ""),
         # Text
         ("scikit-learn/adult-census-income", None, ""),
+        ("jlh/uci-mushrooms", None, ""),
         # Mocked
         # #Image
         ("cifar100", "test", ""),

diff --git a/datasets/flwr_datasets/utils.py b/datasets/flwr_datasets/utils.py
@@ -34,6 +34,7 @@
     "svhn",
     "sentiment140",
     "speech_commands",
+    "jlh/uci-mushrooms",
     "Mike0307/MNIST-M",
     "flwrlabs/usps",
 ]