test errors

baselines/fedht/fedht/aggregate.py

@@ -368,7 +368,7 @@ def _aggregate_n_closest_weights(
 # calls hardthreshold function for each list element in weights_all
 def hardthreshold_list(weights_all, num_keep: int) -> NDArrays:
     """Call hardthreshold."""
-    params = [hardthreshold(each, num_keep) for each in weights_all]
+    params: NDArrays = [hardthreshold(each, num_keep) for each in weights_all]
     return params
 
 
@@ -461,8 +461,7 @@ def aggregate_hardthreshold(
         reduce(np.add, layer_updates) / num_examples_total
         for layer_updates in zip(*weighted_weights2)
     ]
-    params = [hardthreshold_list(layer_updates, num_keep) for layer_updates in hold]
-
-    result: NDArrays = params
+
+    result: NDArrays = [hardthreshold_list(layer_updates, num_keep) for layer_updates in hold]
 
     return result

baselines/fedht/fedht/main.py

@@ -88,7 +88,7 @@ def main(cfg: DictConfig):
     # initialize global model to all zeros
     weights = np.zeros((num_classes, num_features))
     bias = np.zeros(num_classes)
-    init_params: NDArrays = (weights, bias)
+    init_params: NDArrays = [weights, bias]
     init_params = ndarrays_to_parameters(init_params)
 
     # define strategy: fedht

baselines/fedht/fedht/model.py

@@ -78,4 +78,5 @@ def test(model, testloader: DataLoader) -> None:
 
     loss /= len(testloader)
     accuracy = correct / total
+
     return loss, accuracy

baselines/fedht/pyproject.toml

@@ -14,7 +14,7 @@ repository = "https://github.com/adap/flower"
 documentation = "https://flower.ai"
 
 [tool.poetry.dependencies]
-python = ">=3.10.0, <3.11.0"  
+python = ">=3.10.0,<4.0"  
 flwr = { extras = ["simulation"], version = ">=1.11.0" }
 flwr-datasets = { extras = ["vision"], version = ">=0.3.0" }
 scikit-learn =">=1.1.1"