feat(datasets) Add tests for iris dataset (#3967)

adap · Aug 13, 2024 · f5bb435 · f5bb435
1 parent b6f2e2b
commit f5bb435
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 1 deletion.
diff --git a/datasets/flwr_datasets/federated_dataset_test.py b/datasets/flwr_datasets/federated_dataset_test.py
@@ -48,9 +48,10 @@
         ("zh-plus/tiny-imagenet", "valid", ""),
         ("Mike0307/MNIST-M", "test", ""),
         ("flwrlabs/usps", "test", ""),
-        # Text
+        # Tabular
         ("scikit-learn/adult-census-income", None, ""),
         ("jlh/uci-mushrooms", None, ""),
+        ("scikit-learn/iris", None, ""),
         # Mocked
         # #Image
         ("cifar100", "test", ""),

diff --git a/datasets/flwr_datasets/utils.py b/datasets/flwr_datasets/utils.py
@@ -47,6 +47,7 @@
     "jlh/uci-mushrooms",
     "Mike0307/MNIST-M",
     "flwrlabs/usps",
+    "scikit-learn/iris",
 ]