feat(datasets) Add tests for usps dataset (#3832)

adap · Jul 18, 2024 · 4b94caa · 4b94caa
1 parent 02b1959
commit 4b94caa
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", ""),
+        ("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
@@ -34,6 +34,7 @@
     "svhn",
     "sentiment140",
     "speech_commands",
+    "flwrlabs/usps",
 ]
-Original file line number
+Diff line change
@@ Expand Up / @@ -34,6 +34,7 @@ @@
         "svhn",
         "sentiment140",
         "speech_commands",
+        "flwrlabs/usps",
     ]
@@ Expand Down @@