diff --git a/datasets/flwr_datasets/federated_dataset.py b/datasets/flwr_datasets/federated_dataset.py index be03f0a4fcc1..76d434b8140e 100644 --- a/datasets/flwr_datasets/federated_dataset.py +++ b/datasets/flwr_datasets/federated_dataset.py @@ -14,6 +14,7 @@ # ============================================================================== """FederatedDataset.""" + from typing import Dict, Optional import datasets