From 05dcca2ab5d22f9f04e8add7d826e5d38c74e338 Mon Sep 17 00:00:00 2001 From: Adam Narozniak Date: Fri, 21 Jun 2024 14:02:27 +0200 Subject: [PATCH] Drop .keys() when iterating over a dict --- datasets/flwr_datasets/federated_dataset.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/datasets/flwr_datasets/federated_dataset.py b/datasets/flwr_datasets/federated_dataset.py index df92b79b90e4..069f2d5071b9 100644 --- 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(