feat(framework): Bump ruff version to 0.14.5

datasets/flwr_datasets/federated_dataset.py

@@ -132,7 +132,7 @@ 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},
+            "load_partition": dict.fromkeys(self._partitioners, False),
         }
         self._load_dataset_kwargs = load_dataset_kwargs
 
@@ -323,7 +323,7 @@ def _prepare_dataset(self) -> None:
         if self._preprocessor:
             self._dataset = self._preprocessor(self._dataset)
         available_splits = list(self._dataset.keys())
-        self._event["load_split"] = {split: False for split in available_splits}
+        self._event["load_split"] = dict.fromkeys(available_splits, False)
         self._dataset_prepared = True
 
     def _check_if_no_split_keyword_possible(self) -> None:

datasets/flwr_datasets/partitioner/distribution_partitioner.py

@@ -306,7 +306,7 @@ def _determine_partition_id_to_indices_if_needed(  # pylint: disable=R0914
         # Values are the smallest indices of each array in `label_samples`
         # which will be sampled next. Once a sample is taken from a label/key,
         # increment the value (index) by 1.
-        index_tracker = {k: 0 for k in unique_labels}
+        index_tracker = dict.fromkeys(unique_labels, 0)
 
         # Prepare data structure to store indices assigned to partition ids
         self._partition_id_to_indices = {

datasets/pyproject.toml

@@ -76,7 +76,7 @@ pylint = "==3.3.1"
 parameterized = "==0.9.0"
 pytest = "==7.4.4"
 pytest-watcher = "==0.4.3"
-ruff = "==0.4.5"
+ruff = "==0.14.5"
 devtool = [{ path = "./devtool", develop = true }]
 
 [tool.isort]

framework/py/flwr/common/serde_test.py

@@ -190,15 +190,15 @@ def get_str(self, length: int | None = None) -> str:
     def get_value(self, dtype: type[T] | str) -> T:
         """Create a value of a given type."""
         ret: Any = None
-        if dtype == bool:
+        if dtype is bool:
             ret = self.rng.random() < 0.5
-        elif dtype == str:
+        elif dtype is str:
             ret = self.get_str(self.rng.randint(10, 100))
-        elif dtype == int:
+        elif dtype is int:
             ret = self.rng.randint(-1 << 63, (1 << 63) - 1)
-        elif dtype == float:
+        elif dtype is float:
             ret = (self.rng.random() - 0.5) * (2.0 ** self.rng.randint(0, 50))
-        elif dtype == bytes:
+        elif dtype is bytes:
             ret = self.randbytes(self.rng.randint(10, 100))
         elif dtype == "uint":
             ret = self.rng.randint(0, (1 << 64) - 1)

framework/py/flwr/common/serde_utils.py

@@ -114,7 +114,7 @@ def record_value_dict_to_proto(
     Note: `bool` MUST be put in the front of allowd_types if it exists.
     """
     # Move bool to the front
-    if bool in allowed_types and allowed_types[0] != bool:
+    if bool in allowed_types and allowed_types[0] is not bool:
         allowed_types.remove(bool)
         allowed_types.insert(0, bool)
 
@@ -125,7 +125,7 @@ def proto(_v: Any) -> T:
 
 
 def record_value_dict_from_proto(
-    value_dict_proto: MutableMapping[str, Any]
+    value_dict_proto: MutableMapping[str, Any],
 ) -> dict[str, Any]:
     """Deserialize the record value dict from ProtoBuf."""
     return {k: _record_value_from_proto(v) for k, v in value_dict_proto.items()}

framework/py/flwr/server/compat/grid_client_proxy_test.py

@@ -160,7 +160,7 @@ def test_get_properties_and_fail(self) -> None:
 
         # Execute and assert
         self.assertRaises(
-            Exception, self.client.get_properties, ins, timeout=None, group_id=0
+            ValueError, self.client.get_properties, ins, timeout=None, group_id=0
         )
         self._common_assertions(ins)
 
@@ -174,7 +174,7 @@ def test_get_parameters_and_fail(self) -> None:
 
         # Execute and assert
         self.assertRaises(
-            Exception, self.client.get_parameters, ins, timeout=None, group_id=0
+            ValueError, self.client.get_parameters, ins, timeout=None, group_id=0
         )
         self._common_assertions(ins)
 
@@ -188,7 +188,7 @@ def test_fit_and_fail(self) -> None:
         ins = FitIns(parameters, {})
 
         # Execute and assert
-        self.assertRaises(Exception, self.client.fit, ins, timeout=None, group_id=0)
+        self.assertRaises(ValueError, self.client.fit, ins, timeout=None, group_id=0)
         self._common_assertions(ins)
 
     def test_evaluate_and_fail(self) -> None:
@@ -202,7 +202,7 @@ def test_evaluate_and_fail(self) -> None:
 
         # Execute and assert
         self.assertRaises(
-            Exception, self.client.evaluate, ins, timeout=None, group_id=0
+            ValueError, self.client.evaluate, ins, timeout=None, group_id=0
         )
         self._common_assertions(ins)
 

framework/pyproject.toml

@@ -124,7 +124,7 @@ furo = "==2024.8.6"
 sphinx-reredirects = "==0.1.5"
 nbsphinx = "==0.9.5"
 nbstripout = "==0.6.1"
-ruff = "==0.4.5"
+ruff = "==0.14.5"
 sphinx-argparse = "==0.4.0"
 pipreqs = "==0.4.13"
 mdformat = "==0.7.22"