diff --git a/datasets/flwr_datasets/federated_dataset_test.py b/datasets/flwr_datasets/federated_dataset_test.py index 87cb541a430e..ba547efcf4d1 100644 --- a/datasets/flwr_datasets/federated_dataset_test.py +++ b/datasets/flwr_datasets/federated_dataset_test.py @@ -415,6 +415,7 @@ def test_mixed_type_partitioners_creates_from_int(self) -> None: [ ("flwrlabs/femnist", "", "", "writer_id"), ("flwrlabs/ucf101", "test", None, "video_id"), + ("flwrlabs/ambient-acoustic-context", "", None, "speaker_id"), ], ) class NaturalIdPartitionerIntegrationTest(unittest.TestCase):