From 4b94caae038e888e8e34c780b71ecaf256bbd21f Mon Sep 17 00:00:00 2001
From: Adam Narozniak <51029327+adam-narozniak@users.noreply.github.com>
Date: Thu, 18 Jul 2024 17:57:47 +0200
Subject: [PATCH] feat(datasets) Add tests for usps dataset (#3832)

---
 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 bb6d46f266e9..9faf5d93afc9 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", ""),
+        ("flwrlabs/usps", "test", ""),
         # Text
         ("scikit-learn/adult-census-income", None, ""),
         # Mocked
diff --git a/datasets/flwr_datasets/utils.py b/datasets/flwr_datasets/utils.py
index 0ecb96ac9456..2c58340233d9 100644
--- a/datasets/flwr_datasets/utils.py
+++ b/datasets/flwr_datasets/utils.py
@@ -34,6 +34,7 @@
     "svhn",
     "sentiment140",
     "speech_commands",
+    "flwrlabs/usps",
 ]