From ddb1eec04bc3e0345a88f4799718feea96f01574 Mon Sep 17 00:00:00 2001
From: Adam Narozniak <51029327+adam-narozniak@users.noreply.github.com>
Date: Thu, 18 Jul 2024 18:17:25 +0200
Subject: [PATCH] feat(datasets) Add tests for uci-mushrooms dataset (#3841)

---
 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 2f6bdbd6376d..cc3fc6708c36 100644
--- 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
index 00d1f10756ae..b49475baf4e0 100644
--- 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",
 ]