Fix tests

src/py/flwr/driver/driver_client_proxy_test.py

@@ -72,7 +72,9 @@ def test_get_properties(self) -> None:
         )
 
         # Execute
-        value: flwr.common.GetPropertiesRes = client.get_properties(ins, timeout=None)
+        value: flwr.common.GetPropertiesRes = client.get_properties(
+            ins, timeout=None, group_id=None
+        )
 
         # Assert
         assert value.properties["tensor_type"] == "numpy.ndarray"

src/py/flwr/server/fleet/grpc_bidi/ins_scheduler.py

@@ -129,6 +129,7 @@ def _call_client_proxy(
         get_properties_res: GetPropertiesRes = client_proxy.get_properties(
             ins=get_properties_ins,
             timeout=timeout,
+            group_id=None,
         )
         get_properties_res_proto = serde.get_properties_res_to_proto(
             res=get_properties_res
@@ -142,6 +143,7 @@ def _call_client_proxy(
         get_parameters_res: GetParametersRes = client_proxy.get_parameters(
             ins=get_parameters_ins,
             timeout=timeout,
+            group_id=None,
         )
         get_parameters_res_proto = serde.get_parameters_res_to_proto(
             res=get_parameters_res

src/py/flwr/server/server_test.py

@@ -45,13 +45,13 @@ class SuccessClient(ClientProxy):
     """Test class."""
 
     def get_properties(
-        self, ins: GetPropertiesIns, timeout: Optional[float]
+        self, ins: GetPropertiesIns, timeout: Optional[float], group_id: Optional[str]
     ) -> GetPropertiesRes:
         """Raise an Exception because this method is not expected to be called."""
         raise Exception()
 
     def get_parameters(
-        self, ins: GetParametersIns, timeout: Optional[float]
+        self, ins: GetParametersIns, timeout: Optional[float], group_id: Optional[str]
     ) -> GetParametersRes:
         """Raise an Exception because this method is not expected to be called."""
         raise Exception()
@@ -80,7 +80,9 @@ def evaluate(
             metrics={},
         )
 
-    def reconnect(self, ins: ReconnectIns, timeout: Optional[float]) -> DisconnectRes:
+    def reconnect(
+        self, ins: ReconnectIns, timeout: Optional[float], group_id: Optional[str]
+    ) -> DisconnectRes:
         """Simulate reconnect by returning a DisconnectRes with UNKNOWN reason."""
         return DisconnectRes(reason="UNKNOWN")
 
@@ -89,13 +91,13 @@ class FailingClient(ClientProxy):
     """Test class."""
 
     def get_properties(
-        self, ins: GetPropertiesIns, timeout: Optional[float]
+        self, ins: GetPropertiesIns, timeout: Optional[float], group_id: Optional[str]
     ) -> GetPropertiesRes:
         """Raise an Exception to simulate failure in the client."""
         raise Exception()
 
     def get_parameters(
-        self, ins: GetParametersIns, timeout: Optional[float]
+        self, ins: GetParametersIns, timeout: Optional[float], group_id: Optional[str]
     ) -> GetParametersRes:
         """Raise an Exception to simulate failure in the client."""
         raise Exception()
@@ -112,7 +114,9 @@ def evaluate(
         """Raise an Exception to simulate failure in the client."""
         raise Exception()
 
-    def reconnect(self, ins: ReconnectIns, timeout: Optional[float]) -> DisconnectRes:
+    def reconnect(
+        self, ins: ReconnectIns, timeout: Optional[float], group_id: Optional[str]
+    ) -> DisconnectRes:
         """Raise an Exception to simulate failure in the client."""
         raise Exception()