diff --git a/datasets/flwr_datasets/federated_dataset_test.py b/datasets/flwr_datasets/federated_dataset_test.py index 24c59ce029f0..64d75a7a7a5a 100644 --- 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 index 6cea5bd14137..32904ded2861 100644 --- 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", ]