flwr-nightly 1.7.0.dev20240117__py3-none-any.whl → 1.7.0.dev20240119__py3-none-any.whl

flwr/driver/driver_client_proxy.py

@@ -20,7 +20,12 @@ from typing import List, Optional, cast
 
 from flwr import common
 from flwr.common import serde
-from flwr.proto import driver_pb2, node_pb2, task_pb2, transport_pb2
+from flwr.proto import (  # pylint: disable=E0611
+    driver_pb2,
+    node_pb2,
+    task_pb2,
+    transport_pb2,
+)
 from flwr.server.client_proxy import ClientProxy
 
 from .grpc_driver import GrpcDriver
@@ -42,7 +47,7 @@ class DriverClientProxy(ClientProxy):
         self, ins: common.GetPropertiesIns, timeout: Optional[float]
     ) -> common.GetPropertiesRes:
         """Return client's properties."""
-        server_message_proto: transport_pb2.ServerMessage = (
+        server_message_proto: transport_pb2.ServerMessage = (  # pylint: disable=E1101
             serde.server_message_to_proto(
                 server_message=common.ServerMessage(get_properties_ins=ins)
             )
@@ -56,7 +61,7 @@ class DriverClientProxy(ClientProxy):
         self, ins: common.GetParametersIns, timeout: Optional[float]
     ) -> common.GetParametersRes:
         """Return the current local model parameters."""
-        server_message_proto: transport_pb2.ServerMessage = (
+        server_message_proto: transport_pb2.ServerMessage = (  # pylint: disable=E1101
             serde.server_message_to_proto(
                 server_message=common.ServerMessage(get_parameters_ins=ins)
             )
@@ -68,7 +73,7 @@ class DriverClientProxy(ClientProxy):
 
     def fit(self, ins: common.FitIns, timeout: Optional[float]) -> common.FitRes:
         """Train model parameters on the locally held dataset."""
-        server_message_proto: transport_pb2.ServerMessage = (
+        server_message_proto: transport_pb2.ServerMessage = (  # pylint: disable=E1101
             serde.server_message_to_proto(
                 server_message=common.ServerMessage(fit_ins=ins)
             )
@@ -82,7 +87,7 @@ class DriverClientProxy(ClientProxy):
         self, ins: common.EvaluateIns, timeout: Optional[float]
     ) -> common.EvaluateRes:
         """Evaluate model parameters on the locally held dataset."""
-        server_message_proto: transport_pb2.ServerMessage = (
+        server_message_proto: transport_pb2.ServerMessage = (  # pylint: disable=E1101
             serde.server_message_to_proto(
                 server_message=common.ServerMessage(evaluate_ins=ins)
             )
@@ -99,25 +104,29 @@ class DriverClientProxy(ClientProxy):
         return common.DisconnectRes(reason="")  # Nothing to do here (yet)
 
     def _send_receive_msg(
-        self, server_message: transport_pb2.ServerMessage, timeout: Optional[float]
-    ) -> transport_pb2.ClientMessage:
-        task_ins = task_pb2.TaskIns(
+        self,
+        server_message: transport_pb2.ServerMessage,  # pylint: disable=E1101
+        timeout: Optional[float],
+    ) -> transport_pb2.ClientMessage:  # pylint: disable=E1101
+        task_ins = task_pb2.TaskIns(  # pylint: disable=E1101
             task_id="",
             group_id="",
             run_id=self.run_id,
-            task=task_pb2.Task(
-                producer=node_pb2.Node(
+            task=task_pb2.Task(  # pylint: disable=E1101
+                producer=node_pb2.Node(  # pylint: disable=E1101
                     node_id=0,
                     anonymous=True,
                 ),
-                consumer=node_pb2.Node(
+                consumer=node_pb2.Node(  # pylint: disable=E1101
                     node_id=self.node_id,
                     anonymous=self.anonymous,
                 ),
                 legacy_server_message=server_message,
             ),
         )
-        push_task_ins_req = driver_pb2.PushTaskInsRequest(task_ins_list=[task_ins])
+        push_task_ins_req = driver_pb2.PushTaskInsRequest(  # pylint: disable=E1101
+            task_ins_list=[task_ins]
+        )
 
         # Send TaskIns to Driver API
         push_task_ins_res = self.driver.push_task_ins(req=push_task_ins_req)
@@ -133,15 +142,15 @@ class DriverClientProxy(ClientProxy):
             start_time = time.time()
 
         while True:
-            pull_task_res_req = driver_pb2.PullTaskResRequest(
-                node=node_pb2.Node(node_id=0, anonymous=True),
+            pull_task_res_req = driver_pb2.PullTaskResRequest(  # pylint: disable=E1101
+                node=node_pb2.Node(node_id=0, anonymous=True),  # pylint: disable=E1101
                 task_ids=[task_id],
             )
 
             # Ask Driver API for TaskRes
             pull_task_res_res = self.driver.pull_task_res(req=pull_task_res_req)
 
-            task_res_list: List[task_pb2.TaskRes] = list(
+            task_res_list: List[task_pb2.TaskRes] = list(  # pylint: disable=E1101
                 pull_task_res_res.task_res_list
             )
             if len(task_res_list) == 1:

flwr/driver/grpc_driver.py

@@ -23,7 +23,7 @@ import grpc
 from flwr.common import EventType, event
 from flwr.common.grpc import create_channel
 from flwr.common.logger import log
-from flwr.proto.driver_pb2 import (
+from flwr.proto.driver_pb2 import (  # pylint: disable=E0611
     CreateRunRequest,
     CreateRunResponse,
     GetNodesRequest,
@@ -33,7 +33,7 @@ from flwr.proto.driver_pb2 import (
     PushTaskInsRequest,
     PushTaskInsResponse,
 )
-from flwr.proto.driver_pb2_grpc import DriverStub
+from flwr.proto.driver_pb2_grpc import DriverStub  # pylint: disable=E0611
 
 DEFAULT_SERVER_ADDRESS_DRIVER = "[::]:9091"
 

flwr/proto/driver_pb2.py

@@ -1,12 +1,12 @@
 # -*- coding: utf-8 -*-
 # Generated by the protocol buffer compiler.  DO NOT EDIT!
 # source: flwr/proto/driver.proto
+# Protobuf Python Version: 4.25.0
 """Generated protocol buffer code."""
 from google.protobuf import descriptor as _descriptor
 from google.protobuf import descriptor_pool as _descriptor_pool
-from google.protobuf import message as _message
-from google.protobuf import reflection as _reflection
 from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
 # @@protoc_insertion_point(imports)
 
 _sym_db = _symbol_database.Default()
@@ -18,92 +18,27 @@ from flwr.proto import task_pb2 as flwr_dot_proto_dot_task__pb2
 
 DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x17\x66lwr/proto/driver.proto\x12\nflwr.proto\x1a\x15\x66lwr/proto/node.proto\x1a\x15\x66lwr/proto/task.proto\"\x12\n\x10\x43reateRunRequest\"#\n\x11\x43reateRunResponse\x12\x0e\n\x06run_id\x18\x01 \x01(\x12\"!\n\x0fGetNodesRequest\x12\x0e\n\x06run_id\x18\x01 \x01(\x12\"3\n\x10GetNodesResponse\x12\x1f\n\x05nodes\x18\x01 \x03(\x0b\x32\x10.flwr.proto.Node\"@\n\x12PushTaskInsRequest\x12*\n\rtask_ins_list\x18\x01 \x03(\x0b\x32\x13.flwr.proto.TaskIns\"\'\n\x13PushTaskInsResponse\x12\x10\n\x08task_ids\x18\x02 \x03(\t\"F\n\x12PullTaskResRequest\x12\x1e\n\x04node\x18\x01 \x01(\x0b\x32\x10.flwr.proto.Node\x12\x10\n\x08task_ids\x18\x02 \x03(\t\"A\n\x13PullTaskResResponse\x12*\n\rtask_res_list\x18\x01 \x03(\x0b\x32\x13.flwr.proto.TaskRes2\xc1\x02\n\x06\x44river\x12J\n\tCreateRun\x12\x1c.flwr.proto.CreateRunRequest\x1a\x1d.flwr.proto.CreateRunResponse\"\x00\x12G\n\x08GetNodes\x12\x1b.flwr.proto.GetNodesRequest\x1a\x1c.flwr.proto.GetNodesResponse\"\x00\x12P\n\x0bPushTaskIns\x12\x1e.flwr.proto.PushTaskInsRequest\x1a\x1f.flwr.proto.PushTaskInsResponse\"\x00\x12P\n\x0bPullTaskRes\x12\x1e.flwr.proto.PullTaskResRequest\x1a\x1f.flwr.proto.PullTaskResResponse\"\x00\x62\x06proto3')
 
-
-
-_CREATERUNREQUEST = DESCRIPTOR.message_types_by_name['CreateRunRequest']
-_CREATERUNRESPONSE = DESCRIPTOR.message_types_by_name['CreateRunResponse']
-_GETNODESREQUEST = DESCRIPTOR.message_types_by_name['GetNodesRequest']
-_GETNODESRESPONSE = DESCRIPTOR.message_types_by_name['GetNodesResponse']
-_PUSHTASKINSREQUEST = DESCRIPTOR.message_types_by_name['PushTaskInsRequest']
-_PUSHTASKINSRESPONSE = DESCRIPTOR.message_types_by_name['PushTaskInsResponse']
-_PULLTASKRESREQUEST = DESCRIPTOR.message_types_by_name['PullTaskResRequest']
-_PULLTASKRESRESPONSE = DESCRIPTOR.message_types_by_name['PullTaskResResponse']
-CreateRunRequest = _reflection.GeneratedProtocolMessageType('CreateRunRequest', (_message.Message,), {
-  'DESCRIPTOR' : _CREATERUNREQUEST,
-  '__module__' : 'flwr.proto.driver_pb2'
-  # @@protoc_insertion_point(class_scope:flwr.proto.CreateRunRequest)
-  })
-_sym_db.RegisterMessage(CreateRunRequest)
-
-CreateRunResponse = _reflection.GeneratedProtocolMessageType('CreateRunResponse', (_message.Message,), {
-  'DESCRIPTOR' : _CREATERUNRESPONSE,
-  '__module__' : 'flwr.proto.driver_pb2'
-  # @@protoc_insertion_point(class_scope:flwr.proto.CreateRunResponse)
-  })
-_sym_db.RegisterMessage(CreateRunResponse)
-
-GetNodesRequest = _reflection.GeneratedProtocolMessageType('GetNodesRequest', (_message.Message,), {
-  'DESCRIPTOR' : _GETNODESREQUEST,
-  '__module__' : 'flwr.proto.driver_pb2'
-  # @@protoc_insertion_point(class_scope:flwr.proto.GetNodesRequest)
-  })
-_sym_db.RegisterMessage(GetNodesRequest)
-
-GetNodesResponse = _reflection.GeneratedProtocolMessageType('GetNodesResponse', (_message.Message,), {
-  'DESCRIPTOR' : _GETNODESRESPONSE,
-  '__module__' : 'flwr.proto.driver_pb2'
-  # @@protoc_insertion_point(class_scope:flwr.proto.GetNodesResponse)
-  })
-_sym_db.RegisterMessage(GetNodesResponse)
-
-PushTaskInsRequest = _reflection.GeneratedProtocolMessageType('PushTaskInsRequest', (_message.Message,), {
-  'DESCRIPTOR' : _PUSHTASKINSREQUEST,
-  '__module__' : 'flwr.proto.driver_pb2'
-  # @@protoc_insertion_point(class_scope:flwr.proto.PushTaskInsRequest)
-  })
-_sym_db.RegisterMessage(PushTaskInsRequest)
-
-PushTaskInsResponse = _reflection.GeneratedProtocolMessageType('PushTaskInsResponse', (_message.Message,), {
-  'DESCRIPTOR' : _PUSHTASKINSRESPONSE,
-  '__module__' : 'flwr.proto.driver_pb2'
-  # @@protoc_insertion_point(class_scope:flwr.proto.PushTaskInsResponse)
-  })
-_sym_db.RegisterMessage(PushTaskInsResponse)
-
-PullTaskResRequest = _reflection.GeneratedProtocolMessageType('PullTaskResRequest', (_message.Message,), {
-  'DESCRIPTOR' : _PULLTASKRESREQUEST,
-  '__module__' : 'flwr.proto.driver_pb2'
-  # @@protoc_insertion_point(class_scope:flwr.proto.PullTaskResRequest)
-  })
-_sym_db.RegisterMessage(PullTaskResRequest)
-
-PullTaskResResponse = _reflection.GeneratedProtocolMessageType('PullTaskResResponse', (_message.Message,), {
-  'DESCRIPTOR' : _PULLTASKRESRESPONSE,
-  '__module__' : 'flwr.proto.driver_pb2'
-  # @@protoc_insertion_point(class_scope:flwr.proto.PullTaskResResponse)
-  })
-_sym_db.RegisterMessage(PullTaskResResponse)
-
-_DRIVER = DESCRIPTOR.services_by_name['Driver']
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'flwr.proto.driver_pb2', _globals)
 if _descriptor._USE_C_DESCRIPTORS == False:
-
   DESCRIPTOR._options = None
-  _CREATERUNREQUEST._serialized_start=85
-  _CREATERUNREQUEST._serialized_end=103
-  _CREATERUNRESPONSE._serialized_start=105
-  _CREATERUNRESPONSE._serialized_end=140
-  _GETNODESREQUEST._serialized_start=142
-  _GETNODESREQUEST._serialized_end=175
-  _GETNODESRESPONSE._serialized_start=177
-  _GETNODESRESPONSE._serialized_end=228
-  _PUSHTASKINSREQUEST._serialized_start=230
-  _PUSHTASKINSREQUEST._serialized_end=294
-  _PUSHTASKINSRESPONSE._serialized_start=296
-  _PUSHTASKINSRESPONSE._serialized_end=335
-  _PULLTASKRESREQUEST._serialized_start=337
-  _PULLTASKRESREQUEST._serialized_end=407
-  _PULLTASKRESRESPONSE._serialized_start=409
-  _PULLTASKRESRESPONSE._serialized_end=474
-  _DRIVER._serialized_start=477
-  _DRIVER._serialized_end=798
+  _globals['_CREATERUNREQUEST']._serialized_start=85
+  _globals['_CREATERUNREQUEST']._serialized_end=103
+  _globals['_CREATERUNRESPONSE']._serialized_start=105
+  _globals['_CREATERUNRESPONSE']._serialized_end=140
+  _globals['_GETNODESREQUEST']._serialized_start=142
+  _globals['_GETNODESREQUEST']._serialized_end=175
+  _globals['_GETNODESRESPONSE']._serialized_start=177
+  _globals['_GETNODESRESPONSE']._serialized_end=228
+  _globals['_PUSHTASKINSREQUEST']._serialized_start=230
+  _globals['_PUSHTASKINSREQUEST']._serialized_end=294
+  _globals['_PUSHTASKINSRESPONSE']._serialized_start=296
+  _globals['_PUSHTASKINSRESPONSE']._serialized_end=335
+  _globals['_PULLTASKRESREQUEST']._serialized_start=337
+  _globals['_PULLTASKRESREQUEST']._serialized_end=407
+  _globals['_PULLTASKRESRESPONSE']._serialized_start=409
+  _globals['_PULLTASKRESRESPONSE']._serialized_end=474
+  _globals['_DRIVER']._serialized_start=477
+  _globals['_DRIVER']._serialized_end=798
 # @@protoc_insertion_point(module_scope)

flwr/proto/fleet_pb2.py

@@ -1,12 +1,12 @@
 # -*- coding: utf-8 -*-
 # Generated by the protocol buffer compiler.  DO NOT EDIT!
 # source: flwr/proto/fleet.proto
+# Protobuf Python Version: 4.25.0
 """Generated protocol buffer code."""
 from google.protobuf import descriptor as _descriptor
 from google.protobuf import descriptor_pool as _descriptor_pool
-from google.protobuf import message as _message
-from google.protobuf import reflection as _reflection
 from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
 # @@protoc_insertion_point(imports)
 
 _sym_db = _symbol_database.Default()
@@ -18,115 +18,33 @@ from flwr.proto import task_pb2 as flwr_dot_proto_dot_task__pb2
 
 DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x16\x66lwr/proto/fleet.proto\x12\nflwr.proto\x1a\x15\x66lwr/proto/node.proto\x1a\x15\x66lwr/proto/task.proto\"\x13\n\x11\x43reateNodeRequest\"4\n\x12\x43reateNodeResponse\x12\x1e\n\x04node\x18\x01 \x01(\x0b\x32\x10.flwr.proto.Node\"3\n\x11\x44\x65leteNodeRequest\x12\x1e\n\x04node\x18\x01 \x01(\x0b\x32\x10.flwr.proto.Node\"\x14\n\x12\x44\x65leteNodeResponse\"F\n\x12PullTaskInsRequest\x12\x1e\n\x04node\x18\x01 \x01(\x0b\x32\x10.flwr.proto.Node\x12\x10\n\x08task_ids\x18\x02 \x03(\t\"k\n\x13PullTaskInsResponse\x12(\n\treconnect\x18\x01 \x01(\x0b\x32\x15.flwr.proto.Reconnect\x12*\n\rtask_ins_list\x18\x02 \x03(\x0b\x32\x13.flwr.proto.TaskIns\"@\n\x12PushTaskResRequest\x12*\n\rtask_res_list\x18\x01 \x03(\x0b\x32\x13.flwr.proto.TaskRes\"\xae\x01\n\x13PushTaskResResponse\x12(\n\treconnect\x18\x01 \x01(\x0b\x32\x15.flwr.proto.Reconnect\x12=\n\x07results\x18\x02 \x03(\x0b\x32,.flwr.proto.PushTaskResResponse.ResultsEntry\x1a.\n\x0cResultsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12\r\n\x05value\x18\x02 \x01(\r:\x02\x38\x01\"\x1e\n\tReconnect\x12\x11\n\treconnect\x18\x01 \x01(\x04\x32\xc9\x02\n\x05\x46leet\x12M\n\nCreateNode\x12\x1d.flwr.proto.CreateNodeRequest\x1a\x1e.flwr.proto.CreateNodeResponse\"\x00\x12M\n\nDeleteNode\x12\x1d.flwr.proto.DeleteNodeRequest\x1a\x1e.flwr.proto.DeleteNodeResponse\"\x00\x12P\n\x0bPullTaskIns\x12\x1e.flwr.proto.PullTaskInsRequest\x1a\x1f.flwr.proto.PullTaskInsResponse\"\x00\x12P\n\x0bPushTaskRes\x12\x1e.flwr.proto.PushTaskResRequest\x1a\x1f.flwr.proto.PushTaskResResponse\"\x00\x62\x06proto3')
 
-
-
-_CREATENODEREQUEST = DESCRIPTOR.message_types_by_name['CreateNodeRequest']
-_CREATENODERESPONSE = DESCRIPTOR.message_types_by_name['CreateNodeResponse']
-_DELETENODEREQUEST = DESCRIPTOR.message_types_by_name['DeleteNodeRequest']
-_DELETENODERESPONSE = DESCRIPTOR.message_types_by_name['DeleteNodeResponse']
-_PULLTASKINSREQUEST = DESCRIPTOR.message_types_by_name['PullTaskInsRequest']
-_PULLTASKINSRESPONSE = DESCRIPTOR.message_types_by_name['PullTaskInsResponse']
-_PUSHTASKRESREQUEST = DESCRIPTOR.message_types_by_name['PushTaskResRequest']
-_PUSHTASKRESRESPONSE = DESCRIPTOR.message_types_by_name['PushTaskResResponse']
-_PUSHTASKRESRESPONSE_RESULTSENTRY = _PUSHTASKRESRESPONSE.nested_types_by_name['ResultsEntry']
-_RECONNECT = DESCRIPTOR.message_types_by_name['Reconnect']
-CreateNodeRequest = _reflection.GeneratedProtocolMessageType('CreateNodeRequest', (_message.Message,), {
-  'DESCRIPTOR' : _CREATENODEREQUEST,
-  '__module__' : 'flwr.proto.fleet_pb2'
-  # @@protoc_insertion_point(class_scope:flwr.proto.CreateNodeRequest)
-  })
-_sym_db.RegisterMessage(CreateNodeRequest)
-
-CreateNodeResponse = _reflection.GeneratedProtocolMessageType('CreateNodeResponse', (_message.Message,), {
-  'DESCRIPTOR' : _CREATENODERESPONSE,
-  '__module__' : 'flwr.proto.fleet_pb2'
-  # @@protoc_insertion_point(class_scope:flwr.proto.CreateNodeResponse)
-  })
-_sym_db.RegisterMessage(CreateNodeResponse)
-
-DeleteNodeRequest = _reflection.GeneratedProtocolMessageType('DeleteNodeRequest', (_message.Message,), {
-  'DESCRIPTOR' : _DELETENODEREQUEST,
-  '__module__' : 'flwr.proto.fleet_pb2'
-  # @@protoc_insertion_point(class_scope:flwr.proto.DeleteNodeRequest)
-  })
-_sym_db.RegisterMessage(DeleteNodeRequest)
-
-DeleteNodeResponse = _reflection.GeneratedProtocolMessageType('DeleteNodeResponse', (_message.Message,), {
-  'DESCRIPTOR' : _DELETENODERESPONSE,
-  '__module__' : 'flwr.proto.fleet_pb2'
-  # @@protoc_insertion_point(class_scope:flwr.proto.DeleteNodeResponse)
-  })
-_sym_db.RegisterMessage(DeleteNodeResponse)
-
-PullTaskInsRequest = _reflection.GeneratedProtocolMessageType('PullTaskInsRequest', (_message.Message,), {
-  'DESCRIPTOR' : _PULLTASKINSREQUEST,
-  '__module__' : 'flwr.proto.fleet_pb2'
-  # @@protoc_insertion_point(class_scope:flwr.proto.PullTaskInsRequest)
-  })
-_sym_db.RegisterMessage(PullTaskInsRequest)
-
-PullTaskInsResponse = _reflection.GeneratedProtocolMessageType('PullTaskInsResponse', (_message.Message,), {
-  'DESCRIPTOR' : _PULLTASKINSRESPONSE,
-  '__module__' : 'flwr.proto.fleet_pb2'
-  # @@protoc_insertion_point(class_scope:flwr.proto.PullTaskInsResponse)
-  })
-_sym_db.RegisterMessage(PullTaskInsResponse)
-
-PushTaskResRequest = _reflection.GeneratedProtocolMessageType('PushTaskResRequest', (_message.Message,), {
-  'DESCRIPTOR' : _PUSHTASKRESREQUEST,
-  '__module__' : 'flwr.proto.fleet_pb2'
-  # @@protoc_insertion_point(class_scope:flwr.proto.PushTaskResRequest)
-  })
-_sym_db.RegisterMessage(PushTaskResRequest)
-
-PushTaskResResponse = _reflection.GeneratedProtocolMessageType('PushTaskResResponse', (_message.Message,), {
-
-  'ResultsEntry' : _reflection.GeneratedProtocolMessageType('ResultsEntry', (_message.Message,), {
-    'DESCRIPTOR' : _PUSHTASKRESRESPONSE_RESULTSENTRY,
-    '__module__' : 'flwr.proto.fleet_pb2'
-    # @@protoc_insertion_point(class_scope:flwr.proto.PushTaskResResponse.ResultsEntry)
-    })
-  ,
-  'DESCRIPTOR' : _PUSHTASKRESRESPONSE,
-  '__module__' : 'flwr.proto.fleet_pb2'
-  # @@protoc_insertion_point(class_scope:flwr.proto.PushTaskResResponse)
-  })
-_sym_db.RegisterMessage(PushTaskResResponse)
-_sym_db.RegisterMessage(PushTaskResResponse.ResultsEntry)
-
-Reconnect = _reflection.GeneratedProtocolMessageType('Reconnect', (_message.Message,), {
-  'DESCRIPTOR' : _RECONNECT,
-  '__module__' : 'flwr.proto.fleet_pb2'
-  # @@protoc_insertion_point(class_scope:flwr.proto.Reconnect)
-  })
-_sym_db.RegisterMessage(Reconnect)
-
-_FLEET = DESCRIPTOR.services_by_name['Fleet']
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'flwr.proto.fleet_pb2', _globals)
 if _descriptor._USE_C_DESCRIPTORS == False:
-
   DESCRIPTOR._options = None
-  _PUSHTASKRESRESPONSE_RESULTSENTRY._options = None
-  _PUSHTASKRESRESPONSE_RESULTSENTRY._serialized_options = b'8\001'
-  _CREATENODEREQUEST._serialized_start=84
-  _CREATENODEREQUEST._serialized_end=103
-  _CREATENODERESPONSE._serialized_start=105
-  _CREATENODERESPONSE._serialized_end=157
-  _DELETENODEREQUEST._serialized_start=159
-  _DELETENODEREQUEST._serialized_end=210
-  _DELETENODERESPONSE._serialized_start=212
-  _DELETENODERESPONSE._serialized_end=232
-  _PULLTASKINSREQUEST._serialized_start=234
-  _PULLTASKINSREQUEST._serialized_end=304
-  _PULLTASKINSRESPONSE._serialized_start=306
-  _PULLTASKINSRESPONSE._serialized_end=413
-  _PUSHTASKRESREQUEST._serialized_start=415
-  _PUSHTASKRESREQUEST._serialized_end=479
-  _PUSHTASKRESRESPONSE._serialized_start=482
-  _PUSHTASKRESRESPONSE._serialized_end=656
-  _PUSHTASKRESRESPONSE_RESULTSENTRY._serialized_start=610
-  _PUSHTASKRESRESPONSE_RESULTSENTRY._serialized_end=656
-  _RECONNECT._serialized_start=658
-  _RECONNECT._serialized_end=688
-  _FLEET._serialized_start=691
-  _FLEET._serialized_end=1020
+  _globals['_PUSHTASKRESRESPONSE_RESULTSENTRY']._options = None
+  _globals['_PUSHTASKRESRESPONSE_RESULTSENTRY']._serialized_options = b'8\001'
+  _globals['_CREATENODEREQUEST']._serialized_start=84
+  _globals['_CREATENODEREQUEST']._serialized_end=103
+  _globals['_CREATENODERESPONSE']._serialized_start=105
+  _globals['_CREATENODERESPONSE']._serialized_end=157
+  _globals['_DELETENODEREQUEST']._serialized_start=159
+  _globals['_DELETENODEREQUEST']._serialized_end=210
+  _globals['_DELETENODERESPONSE']._serialized_start=212
+  _globals['_DELETENODERESPONSE']._serialized_end=232
+  _globals['_PULLTASKINSREQUEST']._serialized_start=234
+  _globals['_PULLTASKINSREQUEST']._serialized_end=304
+  _globals['_PULLTASKINSRESPONSE']._serialized_start=306
+  _globals['_PULLTASKINSRESPONSE']._serialized_end=413
+  _globals['_PUSHTASKRESREQUEST']._serialized_start=415
+  _globals['_PUSHTASKRESREQUEST']._serialized_end=479
+  _globals['_PUSHTASKRESRESPONSE']._serialized_start=482
+  _globals['_PUSHTASKRESRESPONSE']._serialized_end=656
+  _globals['_PUSHTASKRESRESPONSE_RESULTSENTRY']._serialized_start=610
+  _globals['_PUSHTASKRESRESPONSE_RESULTSENTRY']._serialized_end=656
+  _globals['_RECONNECT']._serialized_start=658
+  _globals['_RECONNECT']._serialized_end=688
+  _globals['_FLEET']._serialized_start=691
+  _globals['_FLEET']._serialized_end=1020
 # @@protoc_insertion_point(module_scope)

flwr/proto/node_pb2.py

@@ -1,12 +1,12 @@
 # -*- coding: utf-8 -*-
 # Generated by the protocol buffer compiler.  DO NOT EDIT!
 # source: flwr/proto/node.proto
+# Protobuf Python Version: 4.25.0
 """Generated protocol buffer code."""
 from google.protobuf import descriptor as _descriptor
 from google.protobuf import descriptor_pool as _descriptor_pool
-from google.protobuf import message as _message
-from google.protobuf import reflection as _reflection
 from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
 # @@protoc_insertion_point(imports)
 
 _sym_db = _symbol_database.Default()
@@ -16,19 +16,11 @@ _sym_db = _symbol_database.Default()
 
 DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x15\x66lwr/proto/node.proto\x12\nflwr.proto\"*\n\x04Node\x12\x0f\n\x07node_id\x18\x01 \x01(\x12\x12\x11\n\tanonymous\x18\x02 \x01(\x08\x62\x06proto3')
 
-
-
-_NODE = DESCRIPTOR.message_types_by_name['Node']
-Node = _reflection.GeneratedProtocolMessageType('Node', (_message.Message,), {
-  'DESCRIPTOR' : _NODE,
-  '__module__' : 'flwr.proto.node_pb2'
-  # @@protoc_insertion_point(class_scope:flwr.proto.Node)
-  })
-_sym_db.RegisterMessage(Node)
-
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'flwr.proto.node_pb2', _globals)
 if _descriptor._USE_C_DESCRIPTORS == False:
-
   DESCRIPTOR._options = None
-  _NODE._serialized_start=37
-  _NODE._serialized_end=79
+  _globals['_NODE']._serialized_start=37
+  _globals['_NODE']._serialized_end=79
 # @@protoc_insertion_point(module_scope)

flwr/proto/task_pb2.py

@@ -1,12 +1,12 @@
 # -*- coding: utf-8 -*-
 # Generated by the protocol buffer compiler.  DO NOT EDIT!
 # source: flwr/proto/task.proto
+# Protobuf Python Version: 4.25.0
 """Generated protocol buffer code."""
 from google.protobuf import descriptor as _descriptor
 from google.protobuf import descriptor_pool as _descriptor_pool
-from google.protobuf import message as _message
-from google.protobuf import reflection as _reflection
 from google.protobuf import symbol_database as _symbol_database
+from google.protobuf.internal import builder as _builder
 # @@protoc_insertion_point(imports)
 
 _sym_db = _symbol_database.Default()
@@ -16,133 +16,39 @@ 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\"\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\"\\\n\x07TaskIns\x12\x0f\n\x07task_id\x18\x01 \x01(\t\x12\x10\n\x08group_id\x18\x02 \x01(\t\x12\x0e\n\x06run_id\x18\x03 \x01(\x12\x12\x1e\n\x04task\x18\x04 \x01(\x0b\x32\x10.flwr.proto.Task\"\\\n\x07TaskRes\x12\x0f\n\x07task_id\x18\x01 \x01(\t\x12\x10\n\x08group_id\x18\x02 \x01(\t\x12\x0e\n\x06run_id\x18\x03 \x01(\x12\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'
-  # @@protoc_insertion_point(class_scope:flwr.proto.Task)
-  })
-_sym_db.RegisterMessage(Task)
-
-TaskIns = _reflection.GeneratedProtocolMessageType('TaskIns', (_message.Message,), {
-  'DESCRIPTOR' : _TASKINS,
-  '__module__' : 'flwr.proto.task_pb2'
-  # @@protoc_insertion_point(class_scope:flwr.proto.TaskIns)
-  })
-_sym_db.RegisterMessage(TaskIns)
-
-TaskRes = _reflection.GeneratedProtocolMessageType('TaskRes', (_message.Message,), {
-  'DESCRIPTOR' : _TASKRES,
-  '__module__' : 'flwr.proto.task_pb2'
-  # @@protoc_insertion_point(class_scope:flwr.proto.TaskRes)
-  })
-_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)
+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\"\xd1\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\x11\n\ttask_type\x18\x07 \x01(\t\x12)\n\x02sa\x18\x08 \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\"\\\n\x07TaskIns\x12\x0f\n\x07task_id\x18\x01 \x01(\t\x12\x10\n\x08group_id\x18\x02 \x01(\t\x12\x0e\n\x06run_id\x18\x03 \x01(\x12\x12\x1e\n\x04task\x18\x04 \x01(\x0b\x32\x10.flwr.proto.Task\"\\\n\x07TaskRes\x12\x0f\n\x07task_id\x18\x01 \x01(\t\x12\x10\n\x08group_id\x18\x02 \x01(\t\x12\x0e\n\x06run_id\x18\x03 \x01(\x12\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')
 
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'flwr.proto.task_pb2', _globals)
 if _descriptor._USE_C_DESCRIPTORS == False:
-
   DESCRIPTOR._options = None
-  _TASK.fields_by_name['legacy_server_message']._options = None
-  _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=407
-  _TASKINS._serialized_start=409
-  _TASKINS._serialized_end=501
-  _TASKRES._serialized_start=503
-  _TASKRES._serialized_end=595
-  _VALUE._serialized_start=598
-  _VALUE._serialized_end=1097
-  _VALUE_DOUBLELIST._serialized_start=953
-  _VALUE_DOUBLELIST._serialized_end=979
-  _VALUE_SINT64LIST._serialized_start=981
-  _VALUE_SINT64LIST._serialized_end=1007
-  _VALUE_BOOLLIST._serialized_start=1009
-  _VALUE_BOOLLIST._serialized_end=1033
-  _VALUE_STRINGLIST._serialized_start=1035
-  _VALUE_STRINGLIST._serialized_end=1061
-  _VALUE_BYTESLIST._serialized_start=1063
-  _VALUE_BYTESLIST._serialized_end=1088
-  _SECUREAGGREGATION._serialized_start=1100
-  _SECUREAGGREGATION._serialized_end=1260
-  _SECUREAGGREGATION_NAMEDVALUESENTRY._serialized_start=1191
-  _SECUREAGGREGATION_NAMEDVALUESENTRY._serialized_end=1260
+  _globals['_TASK'].fields_by_name['legacy_server_message']._options = None
+  _globals['_TASK'].fields_by_name['legacy_server_message']._serialized_options = b'\030\001'
+  _globals['_TASK'].fields_by_name['legacy_client_message']._options = None
+  _globals['_TASK'].fields_by_name['legacy_client_message']._serialized_options = b'\030\001'
+  _globals['_SECUREAGGREGATION_NAMEDVALUESENTRY']._options = None
+  _globals['_SECUREAGGREGATION_NAMEDVALUESENTRY']._serialized_options = b'8\001'
+  _globals['_TASK']._serialized_start=89
+  _globals['_TASK']._serialized_end=426
+  _globals['_TASKINS']._serialized_start=428
+  _globals['_TASKINS']._serialized_end=520
+  _globals['_TASKRES']._serialized_start=522
+  _globals['_TASKRES']._serialized_end=614
+  _globals['_VALUE']._serialized_start=617
+  _globals['_VALUE']._serialized_end=1116
+  _globals['_VALUE_DOUBLELIST']._serialized_start=972
+  _globals['_VALUE_DOUBLELIST']._serialized_end=998
+  _globals['_VALUE_SINT64LIST']._serialized_start=1000
+  _globals['_VALUE_SINT64LIST']._serialized_end=1026
+  _globals['_VALUE_BOOLLIST']._serialized_start=1028
+  _globals['_VALUE_BOOLLIST']._serialized_end=1052
+  _globals['_VALUE_STRINGLIST']._serialized_start=1054
+  _globals['_VALUE_STRINGLIST']._serialized_end=1080
+  _globals['_VALUE_BYTESLIST']._serialized_start=1082
+  _globals['_VALUE_BYTESLIST']._serialized_end=1107
+  _globals['_SECUREAGGREGATION']._serialized_start=1119
+  _globals['_SECUREAGGREGATION']._serialized_end=1279
+  _globals['_SECUREAGGREGATION_NAMEDVALUESENTRY']._serialized_start=1210
+  _globals['_SECUREAGGREGATION_NAMEDVALUESENTRY']._serialized_end=1279
 # @@protoc_insertion_point(module_scope)