flwr-nightly 1.5.0.dev20230608__py3-none-any.whl → 1.5.0.dev20230615__py3-none-any.whl

flwr/client/app.py

@@ -150,7 +150,6 @@ def start_client(
     >>>     root_certificates=Path("/crts/root.pem").read_bytes(),
     >>> )
     """
-
     event(EventType.START_CLIENT_ENTER)
 
     # Parse IP address
@@ -278,7 +277,6 @@ def start_numpy_client(
     >>>     root_certificates=Path("/crts/root.pem").read_bytes(),
     >>> )
     """
-
     # Start
     start_client(
         server_address=server_address,
@@ -321,7 +319,6 @@ def _get_parameters(self: Client, ins: GetParametersIns) -> GetParametersRes:
 
 def _fit(self: Client, ins: FitIns) -> FitRes:
     """Refine the provided parameters using the locally held dataset."""
-
     # Deconstruct FitIns
     parameters: NDArrays = parameters_to_ndarrays(ins.parameters)
 

flwr/client/client.py

@@ -161,7 +161,6 @@ def maybe_call_get_properties(
     client: Client, get_properties_ins: GetPropertiesIns
 ) -> GetPropertiesRes:
     """Call `get_properties` if the client overrides it."""
-
     # Check if client overrides `get_properties`
     if not has_get_properties(client=client):
         # If client does not override `get_properties`, don't call it
@@ -182,7 +181,6 @@ def maybe_call_get_parameters(
     client: Client, get_parameters_ins: GetParametersIns
 ) -> GetParametersRes:
     """Call `get_parameters` if the client overrides it."""
-
     # Check if client overrides `get_parameters`
     if not has_get_parameters(client=client):
         # If client does not override `get_parameters`, don't call it
@@ -201,7 +199,6 @@ def maybe_call_get_parameters(
 
 def maybe_call_fit(client: Client, fit_ins: FitIns) -> FitRes:
     """Call `fit` if the client overrides it."""
-
     # Check if client overrides `fit`
     if not has_fit(client=client):
         # If client does not override `fit`, don't call it
@@ -222,7 +219,6 @@ def maybe_call_fit(client: Client, fit_ins: FitIns) -> FitRes:
 
 def maybe_call_evaluate(client: Client, evaluate_ins: EvaluateIns) -> EvaluateRes:
     """Call `evaluate` if the client overrides it."""
-
     # Check if client overrides `evaluate`
     if not has_evaluate(client=client):
         # If client does not override `evaluate`, don't call it

flwr/client/dpfedavg_numpy_client.py

@@ -33,14 +33,82 @@ class DPFedAvgNumPyClient(NumPyClient):
         self.client = client
 
     def get_properties(self, config: Config) -> Dict[str, Scalar]:
+        """Get client properties using the given Numpy client.
+
+        Parameters
+        ----------
+        config : Config
+            Configuration parameters requested by the server.
+            This can be used to tell the client which properties
+            are needed along with some Scalar attributes.
+
+        Returns
+        -------
+        properties : Dict[str, Scalar]
+            A dictionary mapping arbitrary string keys to values of type
+            bool, bytes, float, int, or str. It can be used to communicate
+            arbitrary property values back to the server.
+        """
         return self.client.get_properties(config)
 
     def get_parameters(self, config: Dict[str, Scalar]) -> NDArrays:
+        """Return the current local model parameters.
+
+        Parameters
+        ----------
+        config : Config
+            Configuration parameters requested by the server.
+            This can be used to tell the client which parameters
+            are needed along with some Scalar attributes.
+
+        Returns
+        -------
+        parameters : NDArrays
+            The local model parameters as a list of NumPy ndarrays.
+        """
         return self.client.get_parameters(config)
 
     def fit(
         self, parameters: NDArrays, config: Dict[str, Scalar]
     ) -> Tuple[NDArrays, int, Dict[str, Scalar]]:
+        """Train the provided parameters using the locally held dataset.
+
+        This method first updates the local model using the original parameters
+        provided. It then calculates the update by subtracting the original
+        parameters from the updated model. The update is then clipped by an L2
+        norm and Gaussian noise is added if specified by the configuration.
+
+        The update is then applied to the original parameters to obtain the
+        updated parameters which are returned along with the number of examples
+        used and metrics computed during the fitting process.
+
+        Parameters
+        ----------
+        parameters : NDArrays
+            The current (global) model parameters.
+        config : Dict[str, Scalar]
+            Configuration parameters which allow the
+            server to influence training on the client. It can be used to
+            communicate arbitrary values from the server to the client, for
+            example, to set the number of (local) training epochs.
+
+        Returns
+        -------
+        parameters : NDArrays
+            The locally updated model parameters.
+        num_examples : int
+            The number of examples used for training.
+        metrics : Dict[str, Scalar]
+            A dictionary mapping arbitrary string keys to values of type
+            bool, bytes, float, int, or str. It can be used to communicate
+            arbitrary values back to the server.
+
+        Raises
+        ------
+        Exception
+            If any required configuration parameters are not provided or are of
+            the wrong type.
+        """
         original_params = copy.deepcopy(parameters)
         # Getting the updated model from the wrapped client
         updated_params, num_examples, metrics = self.client.fit(parameters, config)
@@ -80,4 +148,33 @@ class DPFedAvgNumPyClient(NumPyClient):
     def evaluate(
         self, parameters: NDArrays, config: Dict[str, Scalar]
     ) -> Tuple[float, int, Dict[str, Scalar]]:
+        """Evaluate the provided parameters using the locally held dataset.
+
+        Parameters
+        ----------
+        parameters : NDArrays
+            The current (global) model parameters.
+        config : Dict[str, Scalar]
+            Configuration parameters which allow the server to influence
+            evaluation on the client. It can be used to communicate
+            arbitrary values from the server to the client, for example,
+            to influence the number of examples used for evaluation.
+
+        Returns
+        -------
+        loss : float
+            The evaluation loss of the model on the local dataset.
+        num_examples : int
+            The number of examples used for evaluation.
+        metrics : Dict[str, Scalar]
+            A dictionary mapping arbitrary string keys to values of
+            type bool, bytes, float, int, or str. It can be used to
+            communicate arbitrary values back to the server.
+
+        Warning
+        -------
+        The previous return type format (int, float, float) and the
+        extended format (int, float, float, Dict[str, Scalar]) have been
+        deprecated and removed since Flower 0.19.
+        """
         return self.client.evaluate(parameters, config)

flwr/client/grpc_rere_client/connection.py

@@ -98,7 +98,6 @@ def grpc_request_response(
 
     def receive() -> Optional[ServerMessage]:
         """Receive next task from server."""
-
         # Request instructions (task) from server
         request = PullTaskInsRequest(
             node=Node(node_id=0, anonymous=True),
@@ -121,7 +120,6 @@ def grpc_request_response(
 
     def send(client_message_proto: ClientMessage) -> None:
         """Send task result back to server."""
-
         if state[KEY_TASK_INS] is None:
             log(ERROR, "No current TaskIns")
             return

flwr/client/message_handler/task_handler.py

@@ -26,7 +26,6 @@ def get_server_message(
     pull_task_ins_response: PullTaskInsResponse,
 ) -> Optional[Tuple[TaskIns, ServerMessage]]:
     """Get the first ServerMessage, if available."""
-
     # Extract a single ServerMessage from the response, if possible
     if len(pull_task_ins_response.task_ins_list) == 0:
         return None

flwr/client/numpy_client.py

@@ -25,7 +25,7 @@ class NumPyClient(ABC):
     """Abstract base class for Flower clients using NumPy."""
 
     def get_properties(self, config: Config) -> Dict[str, Scalar]:
-        """Returns a client's set of properties.
+        """Return a client's set of properties.
 
         Parameters
         ----------

flwr/client/rest_client/connection.py

@@ -79,7 +79,6 @@ def http_request_response(
     -------
     receive, send : Callable, Callable
     """
-
     log(
         WARN,
         """
@@ -113,7 +112,6 @@ def http_request_response(
 
     def receive() -> Optional[ServerMessage]:
         """Receive next task from server."""
-
         # Serialize ProtoBuf to bytes
         pull_task_ins_req_proto = PullTaskInsRequest(
             node=Node(node_id=0, anonymous=True),
@@ -170,7 +168,6 @@ def http_request_response(
 
     def send(client_message_proto: ClientMessage) -> None:
         """Send task result back to server."""
-
         if state[KEY_TASK_INS] is None:
             log(ERROR, "No current TaskIns")
             return

flwr/common/address.py

@@ -21,7 +21,7 @@ IPV6: int = 6
 
 
 def parse_address(address: str) -> Optional[Tuple[str, int, Optional[bool]]]:
-    """Parses an IP address into host, port, and version.
+    """Parse an IP address into host, port, and version.
 
     Parameters
     ----------

flwr/common/dp.py

@@ -31,8 +31,7 @@ def _get_update_norm(update: NDArrays) -> float:
 
 
 def add_gaussian_noise(update: NDArrays, std_dev: float) -> NDArrays:
-    """Adds iid Gaussian noise of the given standard deviation to each floating
-    point value in the update."""
+    """Add iid Gaussian noise to each floating point value in the update."""
     update_noised = [
         layer + np.random.normal(0, std_dev, layer.shape) for layer in update
     ]

flwr/common/grpc.py

@@ -31,7 +31,6 @@ def create_channel(
     max_message_length: int = GRPC_MAX_MESSAGE_LENGTH,
 ) -> grpc.Channel:
     """Create a gRPC channel, either secure or insecure."""
-
     # Possible options:
     # https://github.com/grpc/grpc/blob/v1.43.x/include/grpc/impl/codegen/grpc_types.h
     channel_options = [

flwr/common/logger.py

@@ -70,7 +70,6 @@ def configure(
     identifier: str, filename: Optional[str] = None, host: Optional[str] = None
 ) -> None:
     """Configure logging to file and/or remote log server."""
-
     # Create formatter
     string_to_input = f"{identifier} | %(levelname)s %(name)s %(asctime)s "
     string_to_input += "| %(filename)s:%(lineno)d | %(message)s"

flwr/common/serde.py

@@ -458,7 +458,6 @@ def metrics_from_proto(proto: Any) -> typing.Metrics:
 
 def scalar_to_proto(scalar: typing.Scalar) -> Scalar:
     """Serialize `Scalar` to ProtoBuf."""
-
     if isinstance(scalar, bool):
         return Scalar(bool=scalar)
 

flwr/driver/driver.py

@@ -75,7 +75,6 @@ class Driver:
 
     def get_nodes(self, req: driver_pb2.GetNodesRequest) -> driver_pb2.GetNodesResponse:
         """Get client IDs."""
-
         # Check if channel is open
         if self.stub is None:
             log(ERROR, ERROR_MESSAGE_DRIVER_NOT_CONNECTED)
@@ -89,7 +88,6 @@ class Driver:
         self, req: driver_pb2.PushTaskInsRequest
     ) -> driver_pb2.PushTaskInsResponse:
         """Schedule tasks."""
-
         # Check if channel is open
         if self.stub is None:
             log(ERROR, ERROR_MESSAGE_DRIVER_NOT_CONNECTED)
@@ -103,7 +101,6 @@ class Driver:
         self, req: driver_pb2.PullTaskResRequest
     ) -> driver_pb2.PullTaskResResponse:
         """Get task results."""
-
         # Check if channel is open
         if self.stub is None:
             log(ERROR, ERROR_MESSAGE_DRIVER_NOT_CONNECTED)

flwr/driver/driver_client_manager.py

@@ -85,10 +85,12 @@ class DriverClientManager(ClientManager):
         raise NotImplementedError("DriverClientManager.unregister is not implemented")
 
     def all(self) -> Dict[str, ClientProxy]:
+        """Return all available clients."""
         self._update_nodes()
         return self.clients
 
     def wait_for(self, num_clients: int, timeout: int = 86400) -> bool:
+        """Wait until at least `num_clients` are available."""
         start_time = time.time()
         while time.time() < start_time + timeout:
             self._update_nodes()
@@ -103,6 +105,7 @@ class DriverClientManager(ClientManager):
         min_num_clients: Optional[int] = None,
         criterion: Optional[Criterion] = None,
     ) -> List[ClientProxy]:
+        """Sample a number of Flower ClientProxy instances."""
         if min_num_clients is None:
             min_num_clients = num_clients
         self.wait_for(min_num_clients)
@@ -128,6 +131,12 @@ class DriverClientManager(ClientManager):
         return [self.clients[cid] for cid in sampled_cids]
 
     def _update_nodes(self) -> None:
+        """Update the nodes list in the client manager.
+
+        This method communicates with the associated driver to get all node ids. Each
+        node id is then converted into a `DriverClientProxy` instance and stored in the
+        `clients` dictionary with node id as key.
+        """
         get_nodes_res = self.driver.get_nodes(req=driver_pb2.GetNodesRequest())
         all_node_ids = get_nodes_res.node_ids
         for node_id in all_node_ids:

flwr/driver/driver_client_proxy.py

@@ -40,7 +40,7 @@ class DriverClientProxy(ClientProxy):
     def get_properties(
         self, ins: common.GetPropertiesIns, timeout: Optional[float]
     ) -> common.GetPropertiesRes:
-        """Returns client's properties."""
+        """Return client's properties."""
         server_message_proto: transport_pb2.ServerMessage = (
             serde.server_message_to_proto(
                 server_message=common.ServerMessage(get_properties_ins=ins)

flwr/proto/task_pb2.py

@@ -16,13 +16,21 @@ from flwr.proto import node_pb2 as flwr_dot_proto_dot_node__pb2
 from flwr.proto import transport_pb2 as flwr_dot_proto_dot_transport__pb2
 
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x15\x66lwr/proto/task.proto\x12\nflwr.proto\x1a\x15\x66lwr/proto/node.proto\x1a\x1a\x66lwr/proto/transport.proto\"\x93\x02\n\x04Task\x12\"\n\x08producer\x18\x01 \x01(\x0b\x32\x10.flwr.proto.Node\x12\"\n\x08\x63onsumer\x18\x02 \x01(\x0b\x32\x10.flwr.proto.Node\x12\x12\n\ncreated_at\x18\x03 \x01(\t\x12\x14\n\x0c\x64\x65livered_at\x18\x04 \x01(\t\x12\x0b\n\x03ttl\x18\x05 \x01(\t\x12\x10\n\x08\x61ncestry\x18\x06 \x03(\t\x12<\n\x15legacy_server_message\x18\x65 \x01(\x0b\x32\x19.flwr.proto.ServerMessageB\x02\x18\x01\x12<\n\x15legacy_client_message\x18\x66 \x01(\x0b\x32\x19.flwr.proto.ClientMessageB\x02\x18\x01\"a\n\x07TaskIns\x12\x0f\n\x07task_id\x18\x01 \x01(\t\x12\x10\n\x08group_id\x18\x02 \x01(\t\x12\x13\n\x0bworkload_id\x18\x03 \x01(\t\x12\x1e\n\x04task\x18\x04 \x01(\x0b\x32\x10.flwr.proto.Task\"a\n\x07TaskRes\x12\x0f\n\x07task_id\x18\x01 \x01(\t\x12\x10\n\x08group_id\x18\x02 \x01(\t\x12\x13\n\x0bworkload_id\x18\x03 \x01(\t\x12\x1e\n\x04task\x18\x04 \x01(\x0b\x32\x10.flwr.proto.Taskb\x06proto3')
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x15\x66lwr/proto/task.proto\x12\nflwr.proto\x1a\x15\x66lwr/proto/node.proto\x1a\x1a\x66lwr/proto/transport.proto\"\xbe\x02\n\x04Task\x12\"\n\x08producer\x18\x01 \x01(\x0b\x32\x10.flwr.proto.Node\x12\"\n\x08\x63onsumer\x18\x02 \x01(\x0b\x32\x10.flwr.proto.Node\x12\x12\n\ncreated_at\x18\x03 \x01(\t\x12\x14\n\x0c\x64\x65livered_at\x18\x04 \x01(\t\x12\x0b\n\x03ttl\x18\x05 \x01(\t\x12\x10\n\x08\x61ncestry\x18\x06 \x03(\t\x12)\n\x02sa\x18\x07 \x01(\x0b\x32\x1d.flwr.proto.SecureAggregation\x12<\n\x15legacy_server_message\x18\x65 \x01(\x0b\x32\x19.flwr.proto.ServerMessageB\x02\x18\x01\x12<\n\x15legacy_client_message\x18\x66 \x01(\x0b\x32\x19.flwr.proto.ClientMessageB\x02\x18\x01\"a\n\x07TaskIns\x12\x0f\n\x07task_id\x18\x01 \x01(\t\x12\x10\n\x08group_id\x18\x02 \x01(\t\x12\x13\n\x0bworkload_id\x18\x03 \x01(\t\x12\x1e\n\x04task\x18\x04 \x01(\x0b\x32\x10.flwr.proto.Task\"a\n\x07TaskRes\x12\x0f\n\x07task_id\x18\x01 \x01(\t\x12\x10\n\x08group_id\x18\x02 \x01(\t\x12\x13\n\x0bworkload_id\x18\x03 \x01(\t\x12\x1e\n\x04task\x18\x04 \x01(\x0b\x32\x10.flwr.proto.Task\"\xf3\x03\n\x05Value\x12\x10\n\x06\x64ouble\x18\x01 \x01(\x01H\x00\x12\x10\n\x06sint64\x18\x02 \x01(\x12H\x00\x12\x0e\n\x04\x62ool\x18\x03 \x01(\x08H\x00\x12\x10\n\x06string\x18\x04 \x01(\tH\x00\x12\x0f\n\x05\x62ytes\x18\x05 \x01(\x0cH\x00\x12\x33\n\x0b\x64ouble_list\x18\x15 \x01(\x0b\x32\x1c.flwr.proto.Value.DoubleListH\x00\x12\x33\n\x0bsint64_list\x18\x16 \x01(\x0b\x32\x1c.flwr.proto.Value.Sint64ListH\x00\x12/\n\tbool_list\x18\x17 \x01(\x0b\x32\x1a.flwr.proto.Value.BoolListH\x00\x12\x33\n\x0bstring_list\x18\x18 \x01(\x0b\x32\x1c.flwr.proto.Value.StringListH\x00\x12\x31\n\nbytes_list\x18\x19 \x01(\x0b\x32\x1b.flwr.proto.Value.BytesListH\x00\x1a\x1a\n\nDoubleList\x12\x0c\n\x04vals\x18\x01 \x03(\x01\x1a\x1a\n\nSint64List\x12\x0c\n\x04vals\x18\x01 \x03(\x12\x1a\x18\n\x08\x42oolList\x12\x0c\n\x04vals\x18\x01 \x03(\x08\x1a\x1a\n\nStringList\x12\x0c\n\x04vals\x18\x01 \x03(\t\x1a\x19\n\tBytesList\x12\x0c\n\x04vals\x18\x01 \x03(\x0c\x42\x07\n\x05value\"\xa0\x01\n\x11SecureAggregation\x12\x44\n\x0cnamed_values\x18\x01 \x03(\x0b\x32..flwr.proto.SecureAggregation.NamedValuesEntry\x1a\x45\n\x10NamedValuesEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12 \n\x05value\x18\x02 \x01(\x0b\x32\x11.flwr.proto.Value:\x02\x38\x01\x62\x06proto3')
 
 
 
 _TASK = DESCRIPTOR.message_types_by_name['Task']
 _TASKINS = DESCRIPTOR.message_types_by_name['TaskIns']
 _TASKRES = DESCRIPTOR.message_types_by_name['TaskRes']
+_VALUE = DESCRIPTOR.message_types_by_name['Value']
+_VALUE_DOUBLELIST = _VALUE.nested_types_by_name['DoubleList']
+_VALUE_SINT64LIST = _VALUE.nested_types_by_name['Sint64List']
+_VALUE_BOOLLIST = _VALUE.nested_types_by_name['BoolList']
+_VALUE_STRINGLIST = _VALUE.nested_types_by_name['StringList']
+_VALUE_BYTESLIST = _VALUE.nested_types_by_name['BytesList']
+_SECUREAGGREGATION = DESCRIPTOR.message_types_by_name['SecureAggregation']
+_SECUREAGGREGATION_NAMEDVALUESENTRY = _SECUREAGGREGATION.nested_types_by_name['NamedValuesEntry']
 Task = _reflection.GeneratedProtocolMessageType('Task', (_message.Message,), {
   'DESCRIPTOR' : _TASK,
   '__module__' : 'flwr.proto.task_pb2'
@@ -44,6 +52,68 @@ TaskRes = _reflection.GeneratedProtocolMessageType('TaskRes', (_message.Message,
   })
 _sym_db.RegisterMessage(TaskRes)
 
+Value = _reflection.GeneratedProtocolMessageType('Value', (_message.Message,), {
+
+  'DoubleList' : _reflection.GeneratedProtocolMessageType('DoubleList', (_message.Message,), {
+    'DESCRIPTOR' : _VALUE_DOUBLELIST,
+    '__module__' : 'flwr.proto.task_pb2'
+    # @@protoc_insertion_point(class_scope:flwr.proto.Value.DoubleList)
+    })
+  ,
+
+  'Sint64List' : _reflection.GeneratedProtocolMessageType('Sint64List', (_message.Message,), {
+    'DESCRIPTOR' : _VALUE_SINT64LIST,
+    '__module__' : 'flwr.proto.task_pb2'
+    # @@protoc_insertion_point(class_scope:flwr.proto.Value.Sint64List)
+    })
+  ,
+
+  'BoolList' : _reflection.GeneratedProtocolMessageType('BoolList', (_message.Message,), {
+    'DESCRIPTOR' : _VALUE_BOOLLIST,
+    '__module__' : 'flwr.proto.task_pb2'
+    # @@protoc_insertion_point(class_scope:flwr.proto.Value.BoolList)
+    })
+  ,
+
+  'StringList' : _reflection.GeneratedProtocolMessageType('StringList', (_message.Message,), {
+    'DESCRIPTOR' : _VALUE_STRINGLIST,
+    '__module__' : 'flwr.proto.task_pb2'
+    # @@protoc_insertion_point(class_scope:flwr.proto.Value.StringList)
+    })
+  ,
+
+  'BytesList' : _reflection.GeneratedProtocolMessageType('BytesList', (_message.Message,), {
+    'DESCRIPTOR' : _VALUE_BYTESLIST,
+    '__module__' : 'flwr.proto.task_pb2'
+    # @@protoc_insertion_point(class_scope:flwr.proto.Value.BytesList)
+    })
+  ,
+  'DESCRIPTOR' : _VALUE,
+  '__module__' : 'flwr.proto.task_pb2'
+  # @@protoc_insertion_point(class_scope:flwr.proto.Value)
+  })
+_sym_db.RegisterMessage(Value)
+_sym_db.RegisterMessage(Value.DoubleList)
+_sym_db.RegisterMessage(Value.Sint64List)
+_sym_db.RegisterMessage(Value.BoolList)
+_sym_db.RegisterMessage(Value.StringList)
+_sym_db.RegisterMessage(Value.BytesList)
+
+SecureAggregation = _reflection.GeneratedProtocolMessageType('SecureAggregation', (_message.Message,), {
+
+  'NamedValuesEntry' : _reflection.GeneratedProtocolMessageType('NamedValuesEntry', (_message.Message,), {
+    'DESCRIPTOR' : _SECUREAGGREGATION_NAMEDVALUESENTRY,
+    '__module__' : 'flwr.proto.task_pb2'
+    # @@protoc_insertion_point(class_scope:flwr.proto.SecureAggregation.NamedValuesEntry)
+    })
+  ,
+  'DESCRIPTOR' : _SECUREAGGREGATION,
+  '__module__' : 'flwr.proto.task_pb2'
+  # @@protoc_insertion_point(class_scope:flwr.proto.SecureAggregation)
+  })
+_sym_db.RegisterMessage(SecureAggregation)
+_sym_db.RegisterMessage(SecureAggregation.NamedValuesEntry)
+
 if _descriptor._USE_C_DESCRIPTORS == False:
 
   DESCRIPTOR._options = None
@@ -51,10 +121,28 @@ if _descriptor._USE_C_DESCRIPTORS == False:
   _TASK.fields_by_name['legacy_server_message']._serialized_options = b'\030\001'
   _TASK.fields_by_name['legacy_client_message']._options = None
   _TASK.fields_by_name['legacy_client_message']._serialized_options = b'\030\001'
+  _SECUREAGGREGATION_NAMEDVALUESENTRY._options = None
+  _SECUREAGGREGATION_NAMEDVALUESENTRY._serialized_options = b'8\001'
   _TASK._serialized_start=89
-  _TASK._serialized_end=364
-  _TASKINS._serialized_start=366
-  _TASKINS._serialized_end=463
-  _TASKRES._serialized_start=465
-  _TASKRES._serialized_end=562
+  _TASK._serialized_end=407
+  _TASKINS._serialized_start=409
+  _TASKINS._serialized_end=506
+  _TASKRES._serialized_start=508
+  _TASKRES._serialized_end=605
+  _VALUE._serialized_start=608
+  _VALUE._serialized_end=1107
+  _VALUE_DOUBLELIST._serialized_start=963
+  _VALUE_DOUBLELIST._serialized_end=989
+  _VALUE_SINT64LIST._serialized_start=991
+  _VALUE_SINT64LIST._serialized_end=1017
+  _VALUE_BOOLLIST._serialized_start=1019
+  _VALUE_BOOLLIST._serialized_end=1043
+  _VALUE_STRINGLIST._serialized_start=1045
+  _VALUE_STRINGLIST._serialized_end=1071
+  _VALUE_BYTESLIST._serialized_start=1073
+  _VALUE_BYTESLIST._serialized_end=1098
+  _SECUREAGGREGATION._serialized_start=1110
+  _SECUREAGGREGATION._serialized_end=1270
+  _SECUREAGGREGATION_NAMEDVALUESENTRY._serialized_start=1201
+  _SECUREAGGREGATION_NAMEDVALUESENTRY._serialized_end=1270
 # @@protoc_insertion_point(module_scope)

flwr/proto/task_pb2.pyi

@@ -21,6 +21,7 @@ class Task(google.protobuf.message.Message):
     DELIVERED_AT_FIELD_NUMBER: builtins.int
     TTL_FIELD_NUMBER: builtins.int
     ANCESTRY_FIELD_NUMBER: builtins.int
+    SA_FIELD_NUMBER: builtins.int
     LEGACY_SERVER_MESSAGE_FIELD_NUMBER: builtins.int
     LEGACY_CLIENT_MESSAGE_FIELD_NUMBER: builtins.int
     @property
@@ -33,6 +34,8 @@ class Task(google.protobuf.message.Message):
     @property
     def ancestry(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[typing.Text]: ...
     @property
+    def sa(self) -> global___SecureAggregation: ...
+    @property
     def legacy_server_message(self) -> flwr.proto.transport_pb2.ServerMessage: ...
     @property
     def legacy_client_message(self) -> flwr.proto.transport_pb2.ClientMessage: ...
@@ -44,11 +47,12 @@ class Task(google.protobuf.message.Message):
         delivered_at: typing.Text = ...,
         ttl: typing.Text = ...,
         ancestry: typing.Optional[typing.Iterable[typing.Text]] = ...,
+        sa: typing.Optional[global___SecureAggregation] = ...,
         legacy_server_message: typing.Optional[flwr.proto.transport_pb2.ServerMessage] = ...,
         legacy_client_message: typing.Optional[flwr.proto.transport_pb2.ClientMessage] = ...,
         ) -> None: ...
-    def HasField(self, field_name: typing_extensions.Literal["consumer",b"consumer","legacy_client_message",b"legacy_client_message","legacy_server_message",b"legacy_server_message","producer",b"producer"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["ancestry",b"ancestry","consumer",b"consumer","created_at",b"created_at","delivered_at",b"delivered_at","legacy_client_message",b"legacy_client_message","legacy_server_message",b"legacy_server_message","producer",b"producer","ttl",b"ttl"]) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["consumer",b"consumer","legacy_client_message",b"legacy_client_message","legacy_server_message",b"legacy_server_message","producer",b"producer","sa",b"sa"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["ancestry",b"ancestry","consumer",b"consumer","created_at",b"created_at","delivered_at",b"delivered_at","legacy_client_message",b"legacy_client_message","legacy_server_message",b"legacy_server_message","producer",b"producer","sa",b"sa","ttl",b"ttl"]) -> None: ...
 global___Task = Task
 
 class TaskIns(google.protobuf.message.Message):
@@ -94,3 +98,134 @@ class TaskRes(google.protobuf.message.Message):
     def HasField(self, field_name: typing_extensions.Literal["task",b"task"]) -> builtins.bool: ...
     def ClearField(self, field_name: typing_extensions.Literal["group_id",b"group_id","task",b"task","task_id",b"task_id","workload_id",b"workload_id"]) -> None: ...
 global___TaskRes = TaskRes
+
+class Value(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+    class DoubleList(google.protobuf.message.Message):
+        DESCRIPTOR: google.protobuf.descriptor.Descriptor
+        VALS_FIELD_NUMBER: builtins.int
+        @property
+        def vals(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.float]: ...
+        def __init__(self,
+            *,
+            vals: typing.Optional[typing.Iterable[builtins.float]] = ...,
+            ) -> None: ...
+        def ClearField(self, field_name: typing_extensions.Literal["vals",b"vals"]) -> None: ...
+
+    class Sint64List(google.protobuf.message.Message):
+        DESCRIPTOR: google.protobuf.descriptor.Descriptor
+        VALS_FIELD_NUMBER: builtins.int
+        @property
+        def vals(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.int]: ...
+        def __init__(self,
+            *,
+            vals: typing.Optional[typing.Iterable[builtins.int]] = ...,
+            ) -> None: ...
+        def ClearField(self, field_name: typing_extensions.Literal["vals",b"vals"]) -> None: ...
+
+    class BoolList(google.protobuf.message.Message):
+        DESCRIPTOR: google.protobuf.descriptor.Descriptor
+        VALS_FIELD_NUMBER: builtins.int
+        @property
+        def vals(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.bool]: ...
+        def __init__(self,
+            *,
+            vals: typing.Optional[typing.Iterable[builtins.bool]] = ...,
+            ) -> None: ...
+        def ClearField(self, field_name: typing_extensions.Literal["vals",b"vals"]) -> None: ...
+
+    class StringList(google.protobuf.message.Message):
+        DESCRIPTOR: google.protobuf.descriptor.Descriptor
+        VALS_FIELD_NUMBER: builtins.int
+        @property
+        def vals(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[typing.Text]: ...
+        def __init__(self,
+            *,
+            vals: typing.Optional[typing.Iterable[typing.Text]] = ...,
+            ) -> None: ...
+        def ClearField(self, field_name: typing_extensions.Literal["vals",b"vals"]) -> None: ...
+
+    class BytesList(google.protobuf.message.Message):
+        DESCRIPTOR: google.protobuf.descriptor.Descriptor
+        VALS_FIELD_NUMBER: builtins.int
+        @property
+        def vals(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.bytes]: ...
+        def __init__(self,
+            *,
+            vals: typing.Optional[typing.Iterable[builtins.bytes]] = ...,
+            ) -> None: ...
+        def ClearField(self, field_name: typing_extensions.Literal["vals",b"vals"]) -> None: ...
+
+    DOUBLE_FIELD_NUMBER: builtins.int
+    SINT64_FIELD_NUMBER: builtins.int
+    BOOL_FIELD_NUMBER: builtins.int
+    STRING_FIELD_NUMBER: builtins.int
+    BYTES_FIELD_NUMBER: builtins.int
+    DOUBLE_LIST_FIELD_NUMBER: builtins.int
+    SINT64_LIST_FIELD_NUMBER: builtins.int
+    BOOL_LIST_FIELD_NUMBER: builtins.int
+    STRING_LIST_FIELD_NUMBER: builtins.int
+    BYTES_LIST_FIELD_NUMBER: builtins.int
+    double: builtins.float
+    """Single element"""
+
+    sint64: builtins.int
+    bool: builtins.bool
+    string: typing.Text
+    bytes: builtins.bytes
+    @property
+    def double_list(self) -> global___Value.DoubleList:
+        """List types"""
+        pass
+    @property
+    def sint64_list(self) -> global___Value.Sint64List: ...
+    @property
+    def bool_list(self) -> global___Value.BoolList: ...
+    @property
+    def string_list(self) -> global___Value.StringList: ...
+    @property
+    def bytes_list(self) -> global___Value.BytesList: ...
+    def __init__(self,
+        *,
+        double: builtins.float = ...,
+        sint64: builtins.int = ...,
+        bool: builtins.bool = ...,
+        string: typing.Text = ...,
+        bytes: builtins.bytes = ...,
+        double_list: typing.Optional[global___Value.DoubleList] = ...,
+        sint64_list: typing.Optional[global___Value.Sint64List] = ...,
+        bool_list: typing.Optional[global___Value.BoolList] = ...,
+        string_list: typing.Optional[global___Value.StringList] = ...,
+        bytes_list: typing.Optional[global___Value.BytesList] = ...,
+        ) -> None: ...
+    def HasField(self, field_name: typing_extensions.Literal["bool",b"bool","bool_list",b"bool_list","bytes",b"bytes","bytes_list",b"bytes_list","double",b"double","double_list",b"double_list","sint64",b"sint64","sint64_list",b"sint64_list","string",b"string","string_list",b"string_list","value",b"value"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing_extensions.Literal["bool",b"bool","bool_list",b"bool_list","bytes",b"bytes","bytes_list",b"bytes_list","double",b"double","double_list",b"double_list","sint64",b"sint64","sint64_list",b"sint64_list","string",b"string","string_list",b"string_list","value",b"value"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing_extensions.Literal["value",b"value"]) -> typing.Optional[typing_extensions.Literal["double","sint64","bool","string","bytes","double_list","sint64_list","bool_list","string_list","bytes_list"]]: ...
+global___Value = Value
+
+class SecureAggregation(google.protobuf.message.Message):
+    DESCRIPTOR: google.protobuf.descriptor.Descriptor
+    class NamedValuesEntry(google.protobuf.message.Message):
+        DESCRIPTOR: google.protobuf.descriptor.Descriptor
+        KEY_FIELD_NUMBER: builtins.int
+        VALUE_FIELD_NUMBER: builtins.int
+        key: typing.Text
+        @property
+        def value(self) -> global___Value: ...
+        def __init__(self,
+            *,
+            key: typing.Text = ...,
+            value: typing.Optional[global___Value] = ...,
+            ) -> None: ...
+        def HasField(self, field_name: typing_extensions.Literal["value",b"value"]) -> builtins.bool: ...
+        def ClearField(self, field_name: typing_extensions.Literal["key",b"key","value",b"value"]) -> None: ...
+
+    NAMED_VALUES_FIELD_NUMBER: builtins.int
+    @property
+    def named_values(self) -> google.protobuf.internal.containers.MessageMap[typing.Text, global___Value]: ...
+    def __init__(self,
+        *,
+        named_values: typing.Optional[typing.Mapping[typing.Text, global___Value]] = ...,
+        ) -> None: ...
+    def ClearField(self, field_name: typing_extensions.Literal["named_values",b"named_values"]) -> None: ...
+global___SecureAggregation = SecureAggregation