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

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
+    TASK_TYPE_FIELD_NUMBER: builtins.int
     SA_FIELD_NUMBER: builtins.int
     LEGACY_SERVER_MESSAGE_FIELD_NUMBER: builtins.int
     LEGACY_CLIENT_MESSAGE_FIELD_NUMBER: builtins.int
@@ -33,6 +34,7 @@ class Task(google.protobuf.message.Message):
     ttl: typing.Text
     @property
     def ancestry(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[typing.Text]: ...
+    task_type: typing.Text
     @property
     def sa(self) -> global___SecureAggregation: ...
     @property
@@ -47,12 +49,13 @@ class Task(google.protobuf.message.Message):
         delivered_at: typing.Text = ...,
         ttl: typing.Text = ...,
         ancestry: typing.Optional[typing.Iterable[typing.Text]] = ...,
+        task_type: 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","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: ...
+    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","task_type",b"task_type","ttl",b"ttl"]) -> None: ...
 global___Task = Task
 
 class TaskIns(google.protobuf.message.Message):

flwr/proto/transport_pb2.py

@@ -1,13 +1,12 @@
 # -*- coding: utf-8 -*-
 # Generated by the protocol buffer compiler.  DO NOT EDIT!
 # source: flwr/proto/transport.proto
+# Protobuf Python Version: 4.25.0
 """Generated protocol buffer code."""
-from google.protobuf.internal import enum_type_wrapper
 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()
@@ -17,281 +16,73 @@ _sym_db = _symbol_database.Default()
 
 DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1a\x66lwr/proto/transport.proto\x12\nflwr.proto\"9\n\x06Status\x12\x1e\n\x04\x63ode\x18\x01 \x01(\x0e\x32\x10.flwr.proto.Code\x12\x0f\n\x07message\x18\x02 \x01(\t\"2\n\nParameters\x12\x0f\n\x07tensors\x18\x01 \x03(\x0c\x12\x13\n\x0btensor_type\x18\x02 \x01(\t\"\xba\x08\n\rServerMessage\x12?\n\rreconnect_ins\x18\x01 \x01(\x0b\x32&.flwr.proto.ServerMessage.ReconnectInsH\x00\x12H\n\x12get_properties_ins\x18\x02 \x01(\x0b\x32*.flwr.proto.ServerMessage.GetPropertiesInsH\x00\x12H\n\x12get_parameters_ins\x18\x03 \x01(\x0b\x32*.flwr.proto.ServerMessage.GetParametersInsH\x00\x12\x33\n\x07\x66it_ins\x18\x04 \x01(\x0b\x32 .flwr.proto.ServerMessage.FitInsH\x00\x12=\n\x0c\x65valuate_ins\x18\x05 \x01(\x0b\x32%.flwr.proto.ServerMessage.EvaluateInsH\x00\x1a\x1f\n\x0cReconnectIns\x12\x0f\n\x07seconds\x18\x01 \x01(\x03\x1a\x9d\x01\n\x10GetPropertiesIns\x12\x46\n\x06\x63onfig\x18\x01 \x03(\x0b\x32\x36.flwr.proto.ServerMessage.GetPropertiesIns.ConfigEntry\x1a\x41\n\x0b\x43onfigEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12!\n\x05value\x18\x02 \x01(\x0b\x32\x12.flwr.proto.Scalar:\x02\x38\x01\x1a\x9d\x01\n\x10GetParametersIns\x12\x46\n\x06\x63onfig\x18\x01 \x03(\x0b\x32\x36.flwr.proto.ServerMessage.GetParametersIns.ConfigEntry\x1a\x41\n\x0b\x43onfigEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12!\n\x05value\x18\x02 \x01(\x0b\x32\x12.flwr.proto.Scalar:\x02\x38\x01\x1a\xb5\x01\n\x06\x46itIns\x12*\n\nparameters\x18\x01 \x01(\x0b\x32\x16.flwr.proto.Parameters\x12<\n\x06\x63onfig\x18\x02 \x03(\x0b\x32,.flwr.proto.ServerMessage.FitIns.ConfigEntry\x1a\x41\n\x0b\x43onfigEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12!\n\x05value\x18\x02 \x01(\x0b\x32\x12.flwr.proto.Scalar:\x02\x38\x01\x1a\xbf\x01\n\x0b\x45valuateIns\x12*\n\nparameters\x18\x01 \x01(\x0b\x32\x16.flwr.proto.Parameters\x12\x41\n\x06\x63onfig\x18\x02 \x03(\x0b\x32\x31.flwr.proto.ServerMessage.EvaluateIns.ConfigEntry\x1a\x41\n\x0b\x43onfigEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12!\n\x05value\x18\x02 \x01(\x0b\x32\x12.flwr.proto.Scalar:\x02\x38\x01\x42\x05\n\x03msg\"\xa0\t\n\rClientMessage\x12\x41\n\x0e\x64isconnect_res\x18\x01 \x01(\x0b\x32\'.flwr.proto.ClientMessage.DisconnectResH\x00\x12H\n\x12get_properties_res\x18\x02 \x01(\x0b\x32*.flwr.proto.ClientMessage.GetPropertiesResH\x00\x12H\n\x12get_parameters_res\x18\x03 \x01(\x0b\x32*.flwr.proto.ClientMessage.GetParametersResH\x00\x12\x33\n\x07\x66it_res\x18\x04 \x01(\x0b\x32 .flwr.proto.ClientMessage.FitResH\x00\x12=\n\x0c\x65valuate_res\x18\x05 \x01(\x0b\x32%.flwr.proto.ClientMessage.EvaluateResH\x00\x1a\x33\n\rDisconnectRes\x12\"\n\x06reason\x18\x01 \x01(\x0e\x32\x12.flwr.proto.Reason\x1a\xcd\x01\n\x10GetPropertiesRes\x12\"\n\x06status\x18\x01 \x01(\x0b\x32\x12.flwr.proto.Status\x12N\n\nproperties\x18\x02 \x03(\x0b\x32:.flwr.proto.ClientMessage.GetPropertiesRes.PropertiesEntry\x1a\x45\n\x0fPropertiesEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12!\n\x05value\x18\x02 \x01(\x0b\x32\x12.flwr.proto.Scalar:\x02\x38\x01\x1a\x62\n\x10GetParametersRes\x12\"\n\x06status\x18\x01 \x01(\x0b\x32\x12.flwr.proto.Status\x12*\n\nparameters\x18\x02 \x01(\x0b\x32\x16.flwr.proto.Parameters\x1a\xf2\x01\n\x06\x46itRes\x12\"\n\x06status\x18\x01 \x01(\x0b\x32\x12.flwr.proto.Status\x12*\n\nparameters\x18\x02 \x01(\x0b\x32\x16.flwr.proto.Parameters\x12\x14\n\x0cnum_examples\x18\x03 \x01(\x03\x12>\n\x07metrics\x18\x04 \x03(\x0b\x32-.flwr.proto.ClientMessage.FitRes.MetricsEntry\x1a\x42\n\x0cMetricsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12!\n\x05value\x18\x02 \x01(\x0b\x32\x12.flwr.proto.Scalar:\x02\x38\x01\x1a\xde\x01\n\x0b\x45valuateRes\x12\"\n\x06status\x18\x01 \x01(\x0b\x32\x12.flwr.proto.Status\x12\x0c\n\x04loss\x18\x02 \x01(\x02\x12\x14\n\x0cnum_examples\x18\x03 \x01(\x03\x12\x43\n\x07metrics\x18\x04 \x03(\x0b\x32\x32.flwr.proto.ClientMessage.EvaluateRes.MetricsEntry\x1a\x42\n\x0cMetricsEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12!\n\x05value\x18\x02 \x01(\x0b\x32\x12.flwr.proto.Scalar:\x02\x38\x01\x42\x05\n\x03msg\"i\n\x06Scalar\x12\x10\n\x06\x64ouble\x18\x01 \x01(\x01H\x00\x12\x10\n\x06sint64\x18\x08 \x01(\x12H\x00\x12\x0e\n\x04\x62ool\x18\r \x01(\x08H\x00\x12\x10\n\x06string\x18\x0e \x01(\tH\x00\x12\x0f\n\x05\x62ytes\x18\x0f \x01(\x0cH\x00\x42\x08\n\x06scalar*\x8d\x01\n\x04\x43ode\x12\x06\n\x02OK\x10\x00\x12\"\n\x1eGET_PROPERTIES_NOT_IMPLEMENTED\x10\x01\x12\"\n\x1eGET_PARAMETERS_NOT_IMPLEMENTED\x10\x02\x12\x17\n\x13\x46IT_NOT_IMPLEMENTED\x10\x03\x12\x1c\n\x18\x45VALUATE_NOT_IMPLEMENTED\x10\x04*[\n\x06Reason\x12\x0b\n\x07UNKNOWN\x10\x00\x12\r\n\tRECONNECT\x10\x01\x12\x16\n\x12POWER_DISCONNECTED\x10\x02\x12\x14\n\x10WIFI_UNAVAILABLE\x10\x03\x12\x07\n\x03\x41\x43K\x10\x04\x32S\n\rFlowerService\x12\x42\n\x04Join\x12\x19.flwr.proto.ClientMessage\x1a\x19.flwr.proto.ServerMessage\"\x00(\x01\x30\x01\x62\x06proto3')
 
-_CODE = DESCRIPTOR.enum_types_by_name['Code']
-Code = enum_type_wrapper.EnumTypeWrapper(_CODE)
-_REASON = DESCRIPTOR.enum_types_by_name['Reason']
-Reason = enum_type_wrapper.EnumTypeWrapper(_REASON)
-OK = 0
-GET_PROPERTIES_NOT_IMPLEMENTED = 1
-GET_PARAMETERS_NOT_IMPLEMENTED = 2
-FIT_NOT_IMPLEMENTED = 3
-EVALUATE_NOT_IMPLEMENTED = 4
-UNKNOWN = 0
-RECONNECT = 1
-POWER_DISCONNECTED = 2
-WIFI_UNAVAILABLE = 3
-ACK = 4
-
-
-_STATUS = DESCRIPTOR.message_types_by_name['Status']
-_PARAMETERS = DESCRIPTOR.message_types_by_name['Parameters']
-_SERVERMESSAGE = DESCRIPTOR.message_types_by_name['ServerMessage']
-_SERVERMESSAGE_RECONNECTINS = _SERVERMESSAGE.nested_types_by_name['ReconnectIns']
-_SERVERMESSAGE_GETPROPERTIESINS = _SERVERMESSAGE.nested_types_by_name['GetPropertiesIns']
-_SERVERMESSAGE_GETPROPERTIESINS_CONFIGENTRY = _SERVERMESSAGE_GETPROPERTIESINS.nested_types_by_name['ConfigEntry']
-_SERVERMESSAGE_GETPARAMETERSINS = _SERVERMESSAGE.nested_types_by_name['GetParametersIns']
-_SERVERMESSAGE_GETPARAMETERSINS_CONFIGENTRY = _SERVERMESSAGE_GETPARAMETERSINS.nested_types_by_name['ConfigEntry']
-_SERVERMESSAGE_FITINS = _SERVERMESSAGE.nested_types_by_name['FitIns']
-_SERVERMESSAGE_FITINS_CONFIGENTRY = _SERVERMESSAGE_FITINS.nested_types_by_name['ConfigEntry']
-_SERVERMESSAGE_EVALUATEINS = _SERVERMESSAGE.nested_types_by_name['EvaluateIns']
-_SERVERMESSAGE_EVALUATEINS_CONFIGENTRY = _SERVERMESSAGE_EVALUATEINS.nested_types_by_name['ConfigEntry']
-_CLIENTMESSAGE = DESCRIPTOR.message_types_by_name['ClientMessage']
-_CLIENTMESSAGE_DISCONNECTRES = _CLIENTMESSAGE.nested_types_by_name['DisconnectRes']
-_CLIENTMESSAGE_GETPROPERTIESRES = _CLIENTMESSAGE.nested_types_by_name['GetPropertiesRes']
-_CLIENTMESSAGE_GETPROPERTIESRES_PROPERTIESENTRY = _CLIENTMESSAGE_GETPROPERTIESRES.nested_types_by_name['PropertiesEntry']
-_CLIENTMESSAGE_GETPARAMETERSRES = _CLIENTMESSAGE.nested_types_by_name['GetParametersRes']
-_CLIENTMESSAGE_FITRES = _CLIENTMESSAGE.nested_types_by_name['FitRes']
-_CLIENTMESSAGE_FITRES_METRICSENTRY = _CLIENTMESSAGE_FITRES.nested_types_by_name['MetricsEntry']
-_CLIENTMESSAGE_EVALUATERES = _CLIENTMESSAGE.nested_types_by_name['EvaluateRes']
-_CLIENTMESSAGE_EVALUATERES_METRICSENTRY = _CLIENTMESSAGE_EVALUATERES.nested_types_by_name['MetricsEntry']
-_SCALAR = DESCRIPTOR.message_types_by_name['Scalar']
-Status = _reflection.GeneratedProtocolMessageType('Status', (_message.Message,), {
-  'DESCRIPTOR' : _STATUS,
-  '__module__' : 'flwr.proto.transport_pb2'
-  # @@protoc_insertion_point(class_scope:flwr.proto.Status)
-  })
-_sym_db.RegisterMessage(Status)
-
-Parameters = _reflection.GeneratedProtocolMessageType('Parameters', (_message.Message,), {
-  'DESCRIPTOR' : _PARAMETERS,
-  '__module__' : 'flwr.proto.transport_pb2'
-  # @@protoc_insertion_point(class_scope:flwr.proto.Parameters)
-  })
-_sym_db.RegisterMessage(Parameters)
-
-ServerMessage = _reflection.GeneratedProtocolMessageType('ServerMessage', (_message.Message,), {
-
-  'ReconnectIns' : _reflection.GeneratedProtocolMessageType('ReconnectIns', (_message.Message,), {
-    'DESCRIPTOR' : _SERVERMESSAGE_RECONNECTINS,
-    '__module__' : 'flwr.proto.transport_pb2'
-    # @@protoc_insertion_point(class_scope:flwr.proto.ServerMessage.ReconnectIns)
-    })
-  ,
-
-  'GetPropertiesIns' : _reflection.GeneratedProtocolMessageType('GetPropertiesIns', (_message.Message,), {
-
-    'ConfigEntry' : _reflection.GeneratedProtocolMessageType('ConfigEntry', (_message.Message,), {
-      'DESCRIPTOR' : _SERVERMESSAGE_GETPROPERTIESINS_CONFIGENTRY,
-      '__module__' : 'flwr.proto.transport_pb2'
-      # @@protoc_insertion_point(class_scope:flwr.proto.ServerMessage.GetPropertiesIns.ConfigEntry)
-      })
-    ,
-    'DESCRIPTOR' : _SERVERMESSAGE_GETPROPERTIESINS,
-    '__module__' : 'flwr.proto.transport_pb2'
-    # @@protoc_insertion_point(class_scope:flwr.proto.ServerMessage.GetPropertiesIns)
-    })
-  ,
-
-  'GetParametersIns' : _reflection.GeneratedProtocolMessageType('GetParametersIns', (_message.Message,), {
-
-    'ConfigEntry' : _reflection.GeneratedProtocolMessageType('ConfigEntry', (_message.Message,), {
-      'DESCRIPTOR' : _SERVERMESSAGE_GETPARAMETERSINS_CONFIGENTRY,
-      '__module__' : 'flwr.proto.transport_pb2'
-      # @@protoc_insertion_point(class_scope:flwr.proto.ServerMessage.GetParametersIns.ConfigEntry)
-      })
-    ,
-    'DESCRIPTOR' : _SERVERMESSAGE_GETPARAMETERSINS,
-    '__module__' : 'flwr.proto.transport_pb2'
-    # @@protoc_insertion_point(class_scope:flwr.proto.ServerMessage.GetParametersIns)
-    })
-  ,
-
-  'FitIns' : _reflection.GeneratedProtocolMessageType('FitIns', (_message.Message,), {
-
-    'ConfigEntry' : _reflection.GeneratedProtocolMessageType('ConfigEntry', (_message.Message,), {
-      'DESCRIPTOR' : _SERVERMESSAGE_FITINS_CONFIGENTRY,
-      '__module__' : 'flwr.proto.transport_pb2'
-      # @@protoc_insertion_point(class_scope:flwr.proto.ServerMessage.FitIns.ConfigEntry)
-      })
-    ,
-    'DESCRIPTOR' : _SERVERMESSAGE_FITINS,
-    '__module__' : 'flwr.proto.transport_pb2'
-    # @@protoc_insertion_point(class_scope:flwr.proto.ServerMessage.FitIns)
-    })
-  ,
-
-  'EvaluateIns' : _reflection.GeneratedProtocolMessageType('EvaluateIns', (_message.Message,), {
-
-    'ConfigEntry' : _reflection.GeneratedProtocolMessageType('ConfigEntry', (_message.Message,), {
-      'DESCRIPTOR' : _SERVERMESSAGE_EVALUATEINS_CONFIGENTRY,
-      '__module__' : 'flwr.proto.transport_pb2'
-      # @@protoc_insertion_point(class_scope:flwr.proto.ServerMessage.EvaluateIns.ConfigEntry)
-      })
-    ,
-    'DESCRIPTOR' : _SERVERMESSAGE_EVALUATEINS,
-    '__module__' : 'flwr.proto.transport_pb2'
-    # @@protoc_insertion_point(class_scope:flwr.proto.ServerMessage.EvaluateIns)
-    })
-  ,
-  'DESCRIPTOR' : _SERVERMESSAGE,
-  '__module__' : 'flwr.proto.transport_pb2'
-  # @@protoc_insertion_point(class_scope:flwr.proto.ServerMessage)
-  })
-_sym_db.RegisterMessage(ServerMessage)
-_sym_db.RegisterMessage(ServerMessage.ReconnectIns)
-_sym_db.RegisterMessage(ServerMessage.GetPropertiesIns)
-_sym_db.RegisterMessage(ServerMessage.GetPropertiesIns.ConfigEntry)
-_sym_db.RegisterMessage(ServerMessage.GetParametersIns)
-_sym_db.RegisterMessage(ServerMessage.GetParametersIns.ConfigEntry)
-_sym_db.RegisterMessage(ServerMessage.FitIns)
-_sym_db.RegisterMessage(ServerMessage.FitIns.ConfigEntry)
-_sym_db.RegisterMessage(ServerMessage.EvaluateIns)
-_sym_db.RegisterMessage(ServerMessage.EvaluateIns.ConfigEntry)
-
-ClientMessage = _reflection.GeneratedProtocolMessageType('ClientMessage', (_message.Message,), {
-
-  'DisconnectRes' : _reflection.GeneratedProtocolMessageType('DisconnectRes', (_message.Message,), {
-    'DESCRIPTOR' : _CLIENTMESSAGE_DISCONNECTRES,
-    '__module__' : 'flwr.proto.transport_pb2'
-    # @@protoc_insertion_point(class_scope:flwr.proto.ClientMessage.DisconnectRes)
-    })
-  ,
-
-  'GetPropertiesRes' : _reflection.GeneratedProtocolMessageType('GetPropertiesRes', (_message.Message,), {
-
-    'PropertiesEntry' : _reflection.GeneratedProtocolMessageType('PropertiesEntry', (_message.Message,), {
-      'DESCRIPTOR' : _CLIENTMESSAGE_GETPROPERTIESRES_PROPERTIESENTRY,
-      '__module__' : 'flwr.proto.transport_pb2'
-      # @@protoc_insertion_point(class_scope:flwr.proto.ClientMessage.GetPropertiesRes.PropertiesEntry)
-      })
-    ,
-    'DESCRIPTOR' : _CLIENTMESSAGE_GETPROPERTIESRES,
-    '__module__' : 'flwr.proto.transport_pb2'
-    # @@protoc_insertion_point(class_scope:flwr.proto.ClientMessage.GetPropertiesRes)
-    })
-  ,
-
-  'GetParametersRes' : _reflection.GeneratedProtocolMessageType('GetParametersRes', (_message.Message,), {
-    'DESCRIPTOR' : _CLIENTMESSAGE_GETPARAMETERSRES,
-    '__module__' : 'flwr.proto.transport_pb2'
-    # @@protoc_insertion_point(class_scope:flwr.proto.ClientMessage.GetParametersRes)
-    })
-  ,
-
-  'FitRes' : _reflection.GeneratedProtocolMessageType('FitRes', (_message.Message,), {
-
-    'MetricsEntry' : _reflection.GeneratedProtocolMessageType('MetricsEntry', (_message.Message,), {
-      'DESCRIPTOR' : _CLIENTMESSAGE_FITRES_METRICSENTRY,
-      '__module__' : 'flwr.proto.transport_pb2'
-      # @@protoc_insertion_point(class_scope:flwr.proto.ClientMessage.FitRes.MetricsEntry)
-      })
-    ,
-    'DESCRIPTOR' : _CLIENTMESSAGE_FITRES,
-    '__module__' : 'flwr.proto.transport_pb2'
-    # @@protoc_insertion_point(class_scope:flwr.proto.ClientMessage.FitRes)
-    })
-  ,
-
-  'EvaluateRes' : _reflection.GeneratedProtocolMessageType('EvaluateRes', (_message.Message,), {
-
-    'MetricsEntry' : _reflection.GeneratedProtocolMessageType('MetricsEntry', (_message.Message,), {
-      'DESCRIPTOR' : _CLIENTMESSAGE_EVALUATERES_METRICSENTRY,
-      '__module__' : 'flwr.proto.transport_pb2'
-      # @@protoc_insertion_point(class_scope:flwr.proto.ClientMessage.EvaluateRes.MetricsEntry)
-      })
-    ,
-    'DESCRIPTOR' : _CLIENTMESSAGE_EVALUATERES,
-    '__module__' : 'flwr.proto.transport_pb2'
-    # @@protoc_insertion_point(class_scope:flwr.proto.ClientMessage.EvaluateRes)
-    })
-  ,
-  'DESCRIPTOR' : _CLIENTMESSAGE,
-  '__module__' : 'flwr.proto.transport_pb2'
-  # @@protoc_insertion_point(class_scope:flwr.proto.ClientMessage)
-  })
-_sym_db.RegisterMessage(ClientMessage)
-_sym_db.RegisterMessage(ClientMessage.DisconnectRes)
-_sym_db.RegisterMessage(ClientMessage.GetPropertiesRes)
-_sym_db.RegisterMessage(ClientMessage.GetPropertiesRes.PropertiesEntry)
-_sym_db.RegisterMessage(ClientMessage.GetParametersRes)
-_sym_db.RegisterMessage(ClientMessage.FitRes)
-_sym_db.RegisterMessage(ClientMessage.FitRes.MetricsEntry)
-_sym_db.RegisterMessage(ClientMessage.EvaluateRes)
-_sym_db.RegisterMessage(ClientMessage.EvaluateRes.MetricsEntry)
-
-Scalar = _reflection.GeneratedProtocolMessageType('Scalar', (_message.Message,), {
-  'DESCRIPTOR' : _SCALAR,
-  '__module__' : 'flwr.proto.transport_pb2'
-  # @@protoc_insertion_point(class_scope:flwr.proto.Scalar)
-  })
-_sym_db.RegisterMessage(Scalar)
-
-_FLOWERSERVICE = DESCRIPTOR.services_by_name['FlowerService']
+_globals = globals()
+_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
+_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'flwr.proto.transport_pb2', _globals)
 if _descriptor._USE_C_DESCRIPTORS == False:
-
   DESCRIPTOR._options = None
-  _SERVERMESSAGE_GETPROPERTIESINS_CONFIGENTRY._options = None
-  _SERVERMESSAGE_GETPROPERTIESINS_CONFIGENTRY._serialized_options = b'8\001'
-  _SERVERMESSAGE_GETPARAMETERSINS_CONFIGENTRY._options = None
-  _SERVERMESSAGE_GETPARAMETERSINS_CONFIGENTRY._serialized_options = b'8\001'
-  _SERVERMESSAGE_FITINS_CONFIGENTRY._options = None
-  _SERVERMESSAGE_FITINS_CONFIGENTRY._serialized_options = b'8\001'
-  _SERVERMESSAGE_EVALUATEINS_CONFIGENTRY._options = None
-  _SERVERMESSAGE_EVALUATEINS_CONFIGENTRY._serialized_options = b'8\001'
-  _CLIENTMESSAGE_GETPROPERTIESRES_PROPERTIESENTRY._options = None
-  _CLIENTMESSAGE_GETPROPERTIESRES_PROPERTIESENTRY._serialized_options = b'8\001'
-  _CLIENTMESSAGE_FITRES_METRICSENTRY._options = None
-  _CLIENTMESSAGE_FITRES_METRICSENTRY._serialized_options = b'8\001'
-  _CLIENTMESSAGE_EVALUATERES_METRICSENTRY._options = None
-  _CLIENTMESSAGE_EVALUATERES_METRICSENTRY._serialized_options = b'8\001'
-  _CODE._serialized_start=2533
-  _CODE._serialized_end=2674
-  _REASON._serialized_start=2676
-  _REASON._serialized_end=2767
-  _STATUS._serialized_start=42
-  _STATUS._serialized_end=99
-  _PARAMETERS._serialized_start=101
-  _PARAMETERS._serialized_end=151
-  _SERVERMESSAGE._serialized_start=154
-  _SERVERMESSAGE._serialized_end=1236
-  _SERVERMESSAGE_RECONNECTINS._serialized_start=500
-  _SERVERMESSAGE_RECONNECTINS._serialized_end=531
-  _SERVERMESSAGE_GETPROPERTIESINS._serialized_start=534
-  _SERVERMESSAGE_GETPROPERTIESINS._serialized_end=691
-  _SERVERMESSAGE_GETPROPERTIESINS_CONFIGENTRY._serialized_start=626
-  _SERVERMESSAGE_GETPROPERTIESINS_CONFIGENTRY._serialized_end=691
-  _SERVERMESSAGE_GETPARAMETERSINS._serialized_start=694
-  _SERVERMESSAGE_GETPARAMETERSINS._serialized_end=851
-  _SERVERMESSAGE_GETPARAMETERSINS_CONFIGENTRY._serialized_start=626
-  _SERVERMESSAGE_GETPARAMETERSINS_CONFIGENTRY._serialized_end=691
-  _SERVERMESSAGE_FITINS._serialized_start=854
-  _SERVERMESSAGE_FITINS._serialized_end=1035
-  _SERVERMESSAGE_FITINS_CONFIGENTRY._serialized_start=626
-  _SERVERMESSAGE_FITINS_CONFIGENTRY._serialized_end=691
-  _SERVERMESSAGE_EVALUATEINS._serialized_start=1038
-  _SERVERMESSAGE_EVALUATEINS._serialized_end=1229
-  _SERVERMESSAGE_EVALUATEINS_CONFIGENTRY._serialized_start=626
-  _SERVERMESSAGE_EVALUATEINS_CONFIGENTRY._serialized_end=691
-  _CLIENTMESSAGE._serialized_start=1239
-  _CLIENTMESSAGE._serialized_end=2423
-  _CLIENTMESSAGE_DISCONNECTRES._serialized_start=1587
-  _CLIENTMESSAGE_DISCONNECTRES._serialized_end=1638
-  _CLIENTMESSAGE_GETPROPERTIESRES._serialized_start=1641
-  _CLIENTMESSAGE_GETPROPERTIESRES._serialized_end=1846
-  _CLIENTMESSAGE_GETPROPERTIESRES_PROPERTIESENTRY._serialized_start=1777
-  _CLIENTMESSAGE_GETPROPERTIESRES_PROPERTIESENTRY._serialized_end=1846
-  _CLIENTMESSAGE_GETPARAMETERSRES._serialized_start=1848
-  _CLIENTMESSAGE_GETPARAMETERSRES._serialized_end=1946
-  _CLIENTMESSAGE_FITRES._serialized_start=1949
-  _CLIENTMESSAGE_FITRES._serialized_end=2191
-  _CLIENTMESSAGE_FITRES_METRICSENTRY._serialized_start=2125
-  _CLIENTMESSAGE_FITRES_METRICSENTRY._serialized_end=2191
-  _CLIENTMESSAGE_EVALUATERES._serialized_start=2194
-  _CLIENTMESSAGE_EVALUATERES._serialized_end=2416
-  _CLIENTMESSAGE_EVALUATERES_METRICSENTRY._serialized_start=2125
-  _CLIENTMESSAGE_EVALUATERES_METRICSENTRY._serialized_end=2191
-  _SCALAR._serialized_start=2425
-  _SCALAR._serialized_end=2530
-  _FLOWERSERVICE._serialized_start=2769
-  _FLOWERSERVICE._serialized_end=2852
+  _globals['_SERVERMESSAGE_GETPROPERTIESINS_CONFIGENTRY']._options = None
+  _globals['_SERVERMESSAGE_GETPROPERTIESINS_CONFIGENTRY']._serialized_options = b'8\001'
+  _globals['_SERVERMESSAGE_GETPARAMETERSINS_CONFIGENTRY']._options = None
+  _globals['_SERVERMESSAGE_GETPARAMETERSINS_CONFIGENTRY']._serialized_options = b'8\001'
+  _globals['_SERVERMESSAGE_FITINS_CONFIGENTRY']._options = None
+  _globals['_SERVERMESSAGE_FITINS_CONFIGENTRY']._serialized_options = b'8\001'
+  _globals['_SERVERMESSAGE_EVALUATEINS_CONFIGENTRY']._options = None
+  _globals['_SERVERMESSAGE_EVALUATEINS_CONFIGENTRY']._serialized_options = b'8\001'
+  _globals['_CLIENTMESSAGE_GETPROPERTIESRES_PROPERTIESENTRY']._options = None
+  _globals['_CLIENTMESSAGE_GETPROPERTIESRES_PROPERTIESENTRY']._serialized_options = b'8\001'
+  _globals['_CLIENTMESSAGE_FITRES_METRICSENTRY']._options = None
+  _globals['_CLIENTMESSAGE_FITRES_METRICSENTRY']._serialized_options = b'8\001'
+  _globals['_CLIENTMESSAGE_EVALUATERES_METRICSENTRY']._options = None
+  _globals['_CLIENTMESSAGE_EVALUATERES_METRICSENTRY']._serialized_options = b'8\001'
+  _globals['_CODE']._serialized_start=2533
+  _globals['_CODE']._serialized_end=2674
+  _globals['_REASON']._serialized_start=2676
+  _globals['_REASON']._serialized_end=2767
+  _globals['_STATUS']._serialized_start=42
+  _globals['_STATUS']._serialized_end=99
+  _globals['_PARAMETERS']._serialized_start=101
+  _globals['_PARAMETERS']._serialized_end=151
+  _globals['_SERVERMESSAGE']._serialized_start=154
+  _globals['_SERVERMESSAGE']._serialized_end=1236
+  _globals['_SERVERMESSAGE_RECONNECTINS']._serialized_start=500
+  _globals['_SERVERMESSAGE_RECONNECTINS']._serialized_end=531
+  _globals['_SERVERMESSAGE_GETPROPERTIESINS']._serialized_start=534
+  _globals['_SERVERMESSAGE_GETPROPERTIESINS']._serialized_end=691
+  _globals['_SERVERMESSAGE_GETPROPERTIESINS_CONFIGENTRY']._serialized_start=626
+  _globals['_SERVERMESSAGE_GETPROPERTIESINS_CONFIGENTRY']._serialized_end=691
+  _globals['_SERVERMESSAGE_GETPARAMETERSINS']._serialized_start=694
+  _globals['_SERVERMESSAGE_GETPARAMETERSINS']._serialized_end=851
+  _globals['_SERVERMESSAGE_GETPARAMETERSINS_CONFIGENTRY']._serialized_start=626
+  _globals['_SERVERMESSAGE_GETPARAMETERSINS_CONFIGENTRY']._serialized_end=691
+  _globals['_SERVERMESSAGE_FITINS']._serialized_start=854
+  _globals['_SERVERMESSAGE_FITINS']._serialized_end=1035
+  _globals['_SERVERMESSAGE_FITINS_CONFIGENTRY']._serialized_start=626
+  _globals['_SERVERMESSAGE_FITINS_CONFIGENTRY']._serialized_end=691
+  _globals['_SERVERMESSAGE_EVALUATEINS']._serialized_start=1038
+  _globals['_SERVERMESSAGE_EVALUATEINS']._serialized_end=1229
+  _globals['_SERVERMESSAGE_EVALUATEINS_CONFIGENTRY']._serialized_start=626
+  _globals['_SERVERMESSAGE_EVALUATEINS_CONFIGENTRY']._serialized_end=691
+  _globals['_CLIENTMESSAGE']._serialized_start=1239
+  _globals['_CLIENTMESSAGE']._serialized_end=2423
+  _globals['_CLIENTMESSAGE_DISCONNECTRES']._serialized_start=1587
+  _globals['_CLIENTMESSAGE_DISCONNECTRES']._serialized_end=1638
+  _globals['_CLIENTMESSAGE_GETPROPERTIESRES']._serialized_start=1641
+  _globals['_CLIENTMESSAGE_GETPROPERTIESRES']._serialized_end=1846
+  _globals['_CLIENTMESSAGE_GETPROPERTIESRES_PROPERTIESENTRY']._serialized_start=1777
+  _globals['_CLIENTMESSAGE_GETPROPERTIESRES_PROPERTIESENTRY']._serialized_end=1846
+  _globals['_CLIENTMESSAGE_GETPARAMETERSRES']._serialized_start=1848
+  _globals['_CLIENTMESSAGE_GETPARAMETERSRES']._serialized_end=1946
+  _globals['_CLIENTMESSAGE_FITRES']._serialized_start=1949
+  _globals['_CLIENTMESSAGE_FITRES']._serialized_end=2191
+  _globals['_CLIENTMESSAGE_FITRES_METRICSENTRY']._serialized_start=2125
+  _globals['_CLIENTMESSAGE_FITRES_METRICSENTRY']._serialized_end=2191
+  _globals['_CLIENTMESSAGE_EVALUATERES']._serialized_start=2194
+  _globals['_CLIENTMESSAGE_EVALUATERES']._serialized_end=2416
+  _globals['_CLIENTMESSAGE_EVALUATERES_METRICSENTRY']._serialized_start=2125
+  _globals['_CLIENTMESSAGE_EVALUATERES_METRICSENTRY']._serialized_end=2191
+  _globals['_SCALAR']._serialized_start=2425
+  _globals['_SCALAR']._serialized_end=2530
+  _globals['_FLOWERSERVICE']._serialized_start=2769
+  _globals['_FLOWERSERVICE']._serialized_end=2852
 # @@protoc_insertion_point(module_scope)

flwr/server/app.py

@@ -38,9 +38,15 @@ from flwr.common.constant import (
     TRANSPORT_TYPE_REST,
 )
 from flwr.common.logger import log
-from flwr.proto.driver_pb2_grpc import add_DriverServicer_to_server
-from flwr.proto.fleet_pb2_grpc import add_FleetServicer_to_server
-from flwr.proto.transport_pb2_grpc import add_FlowerServiceServicer_to_server
+from flwr.proto.driver_pb2_grpc import (  # pylint: disable=E0611
+    add_DriverServicer_to_server,
+)
+from flwr.proto.fleet_pb2_grpc import (  # pylint: disable=E0611
+    add_FleetServicer_to_server,
+)
+from flwr.proto.transport_pb2_grpc import (  # pylint: disable=E0611
+    add_FlowerServiceServicer_to_server,
+)
 from flwr.server.client_manager import ClientManager, SimpleClientManager
 from flwr.server.driver.driver_servicer import DriverServicer
 from flwr.server.fleet.grpc_bidi.driver_client_manager import DriverClientManager

flwr/server/driver/driver_servicer.py

@@ -22,8 +22,8 @@ from uuid import UUID
 import grpc
 
 from flwr.common.logger import log
-from flwr.proto import driver_pb2_grpc
-from flwr.proto.driver_pb2 import (
+from flwr.proto import driver_pb2_grpc  # pylint: disable=E0611
+from flwr.proto.driver_pb2 import (  # pylint: disable=E0611
     CreateRunRequest,
     CreateRunResponse,
     GetNodesRequest,
@@ -33,8 +33,8 @@ from flwr.proto.driver_pb2 import (
     PushTaskInsRequest,
     PushTaskInsResponse,
 )
-from flwr.proto.node_pb2 import Node
-from flwr.proto.task_pb2 import TaskRes
+from flwr.proto.node_pb2 import Node  # pylint: disable=E0611
+from flwr.proto.task_pb2 import TaskRes  # pylint: disable=E0611
 from flwr.server.state import State, StateFactory
 from flwr.server.utils.validator import validate_task_ins_or_res
 

flwr/server/fleet/grpc_bidi/flower_service_servicer.py

@@ -24,8 +24,11 @@ from typing import Callable, Iterator
 import grpc
 from iterators import TimeoutIterator
 
-from flwr.proto import transport_pb2_grpc
-from flwr.proto.transport_pb2 import ClientMessage, ServerMessage
+from flwr.proto import transport_pb2_grpc  # pylint: disable=E0611
+from flwr.proto.transport_pb2 import (  # pylint: disable=E0611
+    ClientMessage,
+    ServerMessage,
+)
 from flwr.server.client_manager import ClientManager
 from flwr.server.fleet.grpc_bidi.grpc_bridge import GrpcBridge, InsWrapper, ResWrapper
 from flwr.server.fleet.grpc_bidi.grpc_client_proxy import GrpcClientProxy

flwr/server/fleet/grpc_bidi/grpc_bridge.py

@@ -20,7 +20,10 @@ from enum import Enum
 from threading import Condition
 from typing import Iterator, Optional
 
-from flwr.proto.transport_pb2 import ClientMessage, ServerMessage
+from flwr.proto.transport_pb2 import (  # pylint: disable=E0611
+    ClientMessage,
+    ServerMessage,
+)
 
 
 @dataclass

flwr/server/fleet/grpc_bidi/grpc_client_proxy.py

@@ -19,7 +19,10 @@ from typing import Optional
 
 from flwr import common
 from flwr.common import serde
-from flwr.proto.transport_pb2 import ClientMessage, ServerMessage
+from flwr.proto.transport_pb2 import (  # pylint: disable=E0611
+    ClientMessage,
+    ServerMessage,
+)
 from flwr.server.client_proxy import ClientProxy
 from flwr.server.fleet.grpc_bidi.grpc_bridge import GrpcBridge, InsWrapper, ResWrapper
 

flwr/server/fleet/grpc_bidi/grpc_server.py

@@ -24,7 +24,9 @@ import grpc
 
 from flwr.common import GRPC_MAX_MESSAGE_LENGTH
 from flwr.common.logger import log
-from flwr.proto.transport_pb2_grpc import add_FlowerServiceServicer_to_server
+from flwr.proto.transport_pb2_grpc import (  # pylint: disable=E0611
+    add_FlowerServiceServicer_to_server,
+)
 from flwr.server.client_manager import ClientManager
 from flwr.server.driver.driver_servicer import DriverServicer
 from flwr.server.fleet.grpc_bidi.flower_service_servicer import FlowerServiceServicer

flwr/server/fleet/grpc_bidi/ins_scheduler.py

@@ -23,9 +23,12 @@ from typing import Dict, List, Optional
 from flwr.client.message_handler.task_handler import configure_task_res
 from flwr.common import EvaluateRes, FitRes, GetParametersRes, GetPropertiesRes, serde
 from flwr.common.logger import log
-from flwr.proto.node_pb2 import Node
-from flwr.proto.task_pb2 import Task, TaskIns, TaskRes
-from flwr.proto.transport_pb2 import ClientMessage, ServerMessage
+from flwr.proto.node_pb2 import Node  # pylint: disable=E0611
+from flwr.proto.task_pb2 import Task, TaskIns, TaskRes  # pylint: disable=E0611
+from flwr.proto.transport_pb2 import (  # pylint: disable=E0611
+    ClientMessage,
+    ServerMessage,
+)
 from flwr.server.client_proxy import ClientProxy
 from flwr.server.state import State, StateFactory
 

flwr/server/fleet/grpc_rere/fleet_servicer.py

@@ -20,8 +20,8 @@ from logging import INFO
 import grpc
 
 from flwr.common.logger import log
-from flwr.proto import fleet_pb2_grpc
-from flwr.proto.fleet_pb2 import (
+from flwr.proto import fleet_pb2_grpc  # pylint: disable=E0611
+from flwr.proto.fleet_pb2 import (  # pylint: disable=E0611
     CreateNodeRequest,
     CreateNodeResponse,
     DeleteNodeRequest,

flwr/server/fleet/message_handler/message_handler.py

@@ -18,7 +18,7 @@
 from typing import List, Optional
 from uuid import UUID
 
-from flwr.proto.fleet_pb2 import (
+from flwr.proto.fleet_pb2 import (  # pylint: disable=E0611
     CreateNodeRequest,
     CreateNodeResponse,
     DeleteNodeRequest,
@@ -29,8 +29,8 @@ from flwr.proto.fleet_pb2 import (
     PushTaskResResponse,
     Reconnect,
 )
-from flwr.proto.node_pb2 import Node
-from flwr.proto.task_pb2 import TaskIns, TaskRes
+from flwr.proto.node_pb2 import Node  # pylint: disable=E0611
+from flwr.proto.task_pb2 import TaskIns, TaskRes  # pylint: disable=E0611
 from flwr.server.state import State
 
 

flwr/server/fleet/rest_rere/rest_api.py

@@ -18,7 +18,7 @@
 import sys
 
 from flwr.common.constant import MISSING_EXTRA_REST
-from flwr.proto.fleet_pb2 import (
+from flwr.proto.fleet_pb2 import (  # pylint: disable=E0611
     CreateNodeRequest,
     DeleteNodeRequest,
     PullTaskInsRequest,

flwr/server/state/in_memory_state.py

@@ -22,7 +22,7 @@ from typing import Dict, List, Optional, Set
 from uuid import UUID, uuid4
 
 from flwr.common import log, now
-from flwr.proto.task_pb2 import TaskIns, TaskRes
+from flwr.proto.task_pb2 import TaskIns, TaskRes  # pylint: disable=E0611
 from flwr.server.state.state import State
 from flwr.server.utils import validate_task_ins_or_res
 

flwr/server/state/sqlite_state.py

@@ -24,9 +24,12 @@ from typing import Any, Dict, List, Optional, Set, Tuple, Union, cast
 from uuid import UUID, uuid4
 
 from flwr.common import log, now
-from flwr.proto.node_pb2 import Node
-from flwr.proto.task_pb2 import Task, TaskIns, TaskRes
-from flwr.proto.transport_pb2 import ClientMessage, ServerMessage
+from flwr.proto.node_pb2 import Node  # pylint: disable=E0611
+from flwr.proto.task_pb2 import Task, TaskIns, TaskRes  # pylint: disable=E0611
+from flwr.proto.transport_pb2 import (  # pylint: disable=E0611
+    ClientMessage,
+    ServerMessage,
+)
 from flwr.server.utils.validator import validate_task_ins_or_res
 
 from .state import State

flwr/server/state/state.py

@@ -19,7 +19,7 @@ import abc
 from typing import List, Optional, Set
 from uuid import UUID
 
-from flwr.proto.task_pb2 import TaskIns, TaskRes
+from flwr.proto.task_pb2 import TaskIns, TaskRes  # pylint: disable=E0611
 
 
 class State(abc.ABC):

flwr/server/strategy/fedxgb_nn_avg.py

@@ -25,7 +25,7 @@ from logging import WARNING
 from typing import Any, Dict, List, Optional, Tuple, Union
 
 from flwr.common import FitRes, Scalar, ndarrays_to_parameters, parameters_to_ndarrays
-from flwr.common.logger import log
+from flwr.common.logger import log, warn_deprecated_feature
 from flwr.server.client_proxy import ClientProxy
 
 from .aggregate import aggregate
@@ -33,7 +33,13 @@ from .fedavg import FedAvg
 
 
 class FedXgbNnAvg(FedAvg):
-    """Configurable FedXgbNnAvg strategy implementation."""
+    """Configurable FedXgbNnAvg strategy implementation.
+
+    Warning
+    -------
+    This strategy is deprecated, but a copy of it is available in Flower Baselines:
+    https://github.com/adap/flower/tree/main/baselines/hfedxgboost.
+    """
 
     def __init__(self, *args: Any, **kwargs: Any) -> None:
         """Federated XGBoost [Ma et al., 2023] strategy.
@@ -41,6 +47,7 @@ class FedXgbNnAvg(FedAvg):
         Implementation based on https://arxiv.org/abs/2304.07537.
         """
         super().__init__(*args, **kwargs)
+        warn_deprecated_feature("`FedXgbNnAvg` strategy")
 
     def __repr__(self) -> str:
         """Compute a string representation of the strategy."""

flwr/server/utils/validator.py

@@ -17,7 +17,7 @@
 
 from typing import List, Union
 
-from flwr.proto.task_pb2 import TaskIns, TaskRes
+from flwr.proto.task_pb2 import TaskIns, TaskRes  # pylint: disable=E0611
 
 
 # pylint: disable-next=too-many-branches,too-many-statements