Drop .keys() when iterating over a dict

adap · Jun 21, 2024 · 05dcca2 · 05dcca2
1 parent 29dacd8
commit 05dcca2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/datasets/flwr_datasets/federated_dataset.py b/datasets/flwr_datasets/federated_dataset.py
@@ -104,8 +104,8 @@ def __init__(
         # Indicate if the dataset is prepared for `load_partition` or `load_split`
         self._dataset_prepared: bool = False
         self._event = {
-            "load_partition": {split: False for split in self._partitioners.keys()},
-            "load_split": {split: False for split in self._partitioners.keys()},
+            "load_partition": {split: False for split in self._partitioners},
+            "load_split": {split: False for split in self._partitioners},
         }
 
     def load_partition(