flwr 1.22.0__py3-none-any.whl → 1.24.0__py3-none-any.whl

flwr/proto/run_pb2.py

@@ -1,56 +1,65 @@
 # -*- coding: utf-8 -*-
 # Generated by the protocol buffer compiler.  DO NOT EDIT!
+# NO CHECKED-IN PROTOBUF GENCODE
 # source: flwr/proto/run.proto
-# Protobuf Python Version: 4.25.1
+# Protobuf Python Version: 5.29.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 runtime_version as _runtime_version
 from google.protobuf import symbol_database as _symbol_database
 from google.protobuf.internal import builder as _builder
+_runtime_version.ValidateProtobufRuntimeVersion(
+    _runtime_version.Domain.PUBLIC,
+    5,
+    29,
+    0,
+    '',
+    'flwr/proto/run.proto'
+)
 # @@protoc_insertion_point(imports)
 
 _sym_db = _symbol_database.Default()
 
 
-from flwr.proto import fab_pb2 as flwr_dot_proto_dot_fab__pb2
 from flwr.proto import node_pb2 as flwr_dot_proto_dot_node__pb2
 from flwr.proto import recorddict_pb2 as flwr_dot_proto_dot_recorddict__pb2
 from flwr.proto import transport_pb2 as flwr_dot_proto_dot_transport__pb2
 
 
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x14\x66lwr/proto/run.proto\x12\nflwr.proto\x1a\x14\x66lwr/proto/fab.proto\x1a\x15\x66lwr/proto/node.proto\x1a\x1b\x66lwr/proto/recorddict.proto\x1a\x1a\x66lwr/proto/transport.proto\"\xe0\x02\n\x03Run\x12\x0e\n\x06run_id\x18\x01 \x01(\x04\x12\x0e\n\x06\x66\x61\x62_id\x18\x02 \x01(\t\x12\x13\n\x0b\x66\x61\x62_version\x18\x03 \x01(\t\x12<\n\x0foverride_config\x18\x04 \x03(\x0b\x32#.flwr.proto.Run.OverrideConfigEntry\x12\x10\n\x08\x66\x61\x62_hash\x18\x05 \x01(\t\x12\x12\n\npending_at\x18\x06 \x01(\t\x12\x13\n\x0bstarting_at\x18\x07 \x01(\t\x12\x12\n\nrunning_at\x18\x08 \x01(\t\x12\x13\n\x0b\x66inished_at\x18\t \x01(\t\x12%\n\x06status\x18\n \x01(\x0b\x32\x15.flwr.proto.RunStatus\x12\x10\n\x08\x66lwr_aid\x18\x0b \x01(\t\x1aI\n\x13OverrideConfigEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12!\n\x05value\x18\x02 \x01(\x0b\x32\x12.flwr.proto.Scalar:\x02\x38\x01\"@\n\tRunStatus\x12\x0e\n\x06status\x18\x01 \x01(\t\x12\x12\n\nsub_status\x18\x02 \x01(\t\x12\x0f\n\x07\x64\x65tails\x18\x03 \x01(\t\"?\n\rGetRunRequest\x12\x1e\n\x04node\x18\x01 \x01(\x0b\x32\x10.flwr.proto.Node\x12\x0e\n\x06run_id\x18\x02 \x01(\x04\".\n\x0eGetRunResponse\x12\x1c\n\x03run\x18\x01 \x01(\x0b\x32\x0f.flwr.proto.Run\"S\n\x16UpdateRunStatusRequest\x12\x0e\n\x06run_id\x18\x01 \x01(\x04\x12)\n\nrun_status\x18\x02 \x01(\x0b\x32\x15.flwr.proto.RunStatus\"\x19\n\x17UpdateRunStatusResponse\"F\n\x13GetRunStatusRequest\x12\x1e\n\x04node\x18\x01 \x01(\x0b\x32\x10.flwr.proto.Node\x12\x0f\n\x07run_ids\x18\x02 \x03(\x04\"\xb1\x01\n\x14GetRunStatusResponse\x12L\n\x0frun_status_dict\x18\x01 \x03(\x0b\x32\x33.flwr.proto.GetRunStatusResponse.RunStatusDictEntry\x1aK\n\x12RunStatusDictEntry\x12\x0b\n\x03key\x18\x01 \x01(\x04\x12$\n\x05value\x18\x02 \x01(\x0b\x32\x15.flwr.proto.RunStatus:\x02\x38\x01\"-\n\x1bGetFederationOptionsRequest\x12\x0e\n\x06run_id\x18\x01 \x01(\x04\"T\n\x1cGetFederationOptionsResponse\x12\x34\n\x12\x66\x65\x64\x65ration_options\x18\x01 \x01(\x0b\x32\x18.flwr.proto.ConfigRecordb\x06proto3')
+DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x14\x66lwr/proto/run.proto\x12\nflwr.proto\x1a\x15\x66lwr/proto/node.proto\x1a\x1b\x66lwr/proto/recorddict.proto\x1a\x1a\x66lwr/proto/transport.proto\"\xf4\x02\n\x03Run\x12\x0e\n\x06run_id\x18\x01 \x01(\x04\x12\x0e\n\x06\x66\x61\x62_id\x18\x02 \x01(\t\x12\x13\n\x0b\x66\x61\x62_version\x18\x03 \x01(\t\x12<\n\x0foverride_config\x18\x04 \x03(\x0b\x32#.flwr.proto.Run.OverrideConfigEntry\x12\x10\n\x08\x66\x61\x62_hash\x18\x05 \x01(\t\x12\x12\n\npending_at\x18\x06 \x01(\t\x12\x13\n\x0bstarting_at\x18\x07 \x01(\t\x12\x12\n\nrunning_at\x18\x08 \x01(\t\x12\x13\n\x0b\x66inished_at\x18\t \x01(\t\x12%\n\x06status\x18\n \x01(\x0b\x32\x15.flwr.proto.RunStatus\x12\x10\n\x08\x66lwr_aid\x18\x0b \x01(\t\x12\x12\n\nfederation\x18\x0c \x01(\t\x1aI\n\x13OverrideConfigEntry\x12\x0b\n\x03key\x18\x01 \x01(\t\x12!\n\x05value\x18\x02 \x01(\x0b\x32\x12.flwr.proto.Scalar:\x02\x38\x01\"@\n\tRunStatus\x12\x0e\n\x06status\x18\x01 \x01(\t\x12\x12\n\nsub_status\x18\x02 \x01(\t\x12\x0f\n\x07\x64\x65tails\x18\x03 \x01(\t\"?\n\rGetRunRequest\x12\x1e\n\x04node\x18\x01 \x01(\x0b\x32\x10.flwr.proto.Node\x12\x0e\n\x06run_id\x18\x02 \x01(\x04\".\n\x0eGetRunResponse\x12\x1c\n\x03run\x18\x01 \x01(\x0b\x32\x0f.flwr.proto.Run\"S\n\x16UpdateRunStatusRequest\x12\x0e\n\x06run_id\x18\x01 \x01(\x04\x12)\n\nrun_status\x18\x02 \x01(\x0b\x32\x15.flwr.proto.RunStatus\"\x19\n\x17UpdateRunStatusResponse\"F\n\x13GetRunStatusRequest\x12\x1e\n\x04node\x18\x01 \x01(\x0b\x32\x10.flwr.proto.Node\x12\x0f\n\x07run_ids\x18\x02 \x03(\x04\"\xb1\x01\n\x14GetRunStatusResponse\x12L\n\x0frun_status_dict\x18\x01 \x03(\x0b\x32\x33.flwr.proto.GetRunStatusResponse.RunStatusDictEntry\x1aK\n\x12RunStatusDictEntry\x12\x0b\n\x03key\x18\x01 \x01(\x04\x12$\n\x05value\x18\x02 \x01(\x0b\x32\x15.flwr.proto.RunStatus:\x02\x38\x01\"-\n\x1bGetFederationOptionsRequest\x12\x0e\n\x06run_id\x18\x01 \x01(\x04\"T\n\x1cGetFederationOptionsResponse\x12\x34\n\x12\x66\x65\x64\x65ration_options\x18\x01 \x01(\x0b\x32\x18.flwr.proto.ConfigRecordb\x06proto3')
 
 _globals = globals()
 _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
 _builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'flwr.proto.run_pb2', _globals)
-if _descriptor._USE_C_DESCRIPTORS == False:
-  DESCRIPTOR._options = None
-  _globals['_RUN_OVERRIDECONFIGENTRY']._options = None
+if not _descriptor._USE_C_DESCRIPTORS:
+  DESCRIPTOR._loaded_options = None
+  _globals['_RUN_OVERRIDECONFIGENTRY']._loaded_options = None
   _globals['_RUN_OVERRIDECONFIGENTRY']._serialized_options = b'8\001'
-  _globals['_GETRUNSTATUSRESPONSE_RUNSTATUSDICTENTRY']._options = None
+  _globals['_GETRUNSTATUSRESPONSE_RUNSTATUSDICTENTRY']._loaded_options = None
   _globals['_GETRUNSTATUSRESPONSE_RUNSTATUSDICTENTRY']._serialized_options = b'8\001'
-  _globals['_RUN']._serialized_start=139
-  _globals['_RUN']._serialized_end=491
-  _globals['_RUN_OVERRIDECONFIGENTRY']._serialized_start=418
-  _globals['_RUN_OVERRIDECONFIGENTRY']._serialized_end=491
-  _globals['_RUNSTATUS']._serialized_start=493
-  _globals['_RUNSTATUS']._serialized_end=557
-  _globals['_GETRUNREQUEST']._serialized_start=559
-  _globals['_GETRUNREQUEST']._serialized_end=622
-  _globals['_GETRUNRESPONSE']._serialized_start=624
-  _globals['_GETRUNRESPONSE']._serialized_end=670
-  _globals['_UPDATERUNSTATUSREQUEST']._serialized_start=672
-  _globals['_UPDATERUNSTATUSREQUEST']._serialized_end=755
-  _globals['_UPDATERUNSTATUSRESPONSE']._serialized_start=757
-  _globals['_UPDATERUNSTATUSRESPONSE']._serialized_end=782
-  _globals['_GETRUNSTATUSREQUEST']._serialized_start=784
-  _globals['_GETRUNSTATUSREQUEST']._serialized_end=854
-  _globals['_GETRUNSTATUSRESPONSE']._serialized_start=857
-  _globals['_GETRUNSTATUSRESPONSE']._serialized_end=1034
-  _globals['_GETRUNSTATUSRESPONSE_RUNSTATUSDICTENTRY']._serialized_start=959
-  _globals['_GETRUNSTATUSRESPONSE_RUNSTATUSDICTENTRY']._serialized_end=1034
-  _globals['_GETFEDERATIONOPTIONSREQUEST']._serialized_start=1036
-  _globals['_GETFEDERATIONOPTIONSREQUEST']._serialized_end=1081
-  _globals['_GETFEDERATIONOPTIONSRESPONSE']._serialized_start=1083
-  _globals['_GETFEDERATIONOPTIONSRESPONSE']._serialized_end=1167
+  _globals['_RUN']._serialized_start=117
+  _globals['_RUN']._serialized_end=489
+  _globals['_RUN_OVERRIDECONFIGENTRY']._serialized_start=416
+  _globals['_RUN_OVERRIDECONFIGENTRY']._serialized_end=489
+  _globals['_RUNSTATUS']._serialized_start=491
+  _globals['_RUNSTATUS']._serialized_end=555
+  _globals['_GETRUNREQUEST']._serialized_start=557
+  _globals['_GETRUNREQUEST']._serialized_end=620
+  _globals['_GETRUNRESPONSE']._serialized_start=622
+  _globals['_GETRUNRESPONSE']._serialized_end=668
+  _globals['_UPDATERUNSTATUSREQUEST']._serialized_start=670
+  _globals['_UPDATERUNSTATUSREQUEST']._serialized_end=753
+  _globals['_UPDATERUNSTATUSRESPONSE']._serialized_start=755
+  _globals['_UPDATERUNSTATUSRESPONSE']._serialized_end=780
+  _globals['_GETRUNSTATUSREQUEST']._serialized_start=782
+  _globals['_GETRUNSTATUSREQUEST']._serialized_end=852
+  _globals['_GETRUNSTATUSRESPONSE']._serialized_start=855
+  _globals['_GETRUNSTATUSRESPONSE']._serialized_end=1032
+  _globals['_GETRUNSTATUSRESPONSE_RUNSTATUSDICTENTRY']._serialized_start=957
+  _globals['_GETRUNSTATUSRESPONSE_RUNSTATUSDICTENTRY']._serialized_end=1032
+  _globals['_GETFEDERATIONOPTIONSREQUEST']._serialized_start=1034
+  _globals['_GETFEDERATIONOPTIONSREQUEST']._serialized_end=1079
+  _globals['_GETFEDERATIONOPTIONSRESPONSE']._serialized_start=1081
+  _globals['_GETFEDERATIONOPTIONSRESPONSE']._serialized_end=1165
 # @@protoc_insertion_point(module_scope)

flwr/proto/run_pb2.pyi

@@ -1,8 +1,24 @@
 """
 @generated by mypy-protobuf.  Do not edit manually!
 isort:skip_file
+Copyright 2024 Flower Labs GmbH. All Rights Reserved.
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+==============================================================================
 """
+
 import builtins
+import collections.abc
 import flwr.proto.node_pb2
 import flwr.proto.recorddict_pb2
 import flwr.proto.transport_pb2
@@ -10,26 +26,30 @@ import google.protobuf.descriptor
 import google.protobuf.internal.containers
 import google.protobuf.message
 import typing
-import typing_extensions
 
 DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
 
+@typing.final
 class Run(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    @typing.final
     class OverrideConfigEntry(google.protobuf.message.Message):
         DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
         KEY_FIELD_NUMBER: builtins.int
         VALUE_FIELD_NUMBER: builtins.int
-        key: typing.Text
+        key: builtins.str
         @property
         def value(self) -> flwr.proto.transport_pb2.Scalar: ...
-        def __init__(self,
+        def __init__(
+            self,
             *,
-            key: typing.Text = ...,
-            value: typing.Optional[flwr.proto.transport_pb2.Scalar] = ...,
-            ) -> 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: ...
+            key: builtins.str = ...,
+            value: flwr.proto.transport_pb2.Scalar | None = ...,
+        ) -> None: ...
+        def HasField(self, field_name: typing.Literal["value", b"value"]) -> builtins.bool: ...
+        def ClearField(self, field_name: typing.Literal["key", b"key", "value", b"value"]) -> None: ...
 
     RUN_ID_FIELD_NUMBER: builtins.int
     FAB_ID_FIELD_NUMBER: builtins.int
@@ -42,179 +62,224 @@ class Run(google.protobuf.message.Message):
     FINISHED_AT_FIELD_NUMBER: builtins.int
     STATUS_FIELD_NUMBER: builtins.int
     FLWR_AID_FIELD_NUMBER: builtins.int
+    FEDERATION_FIELD_NUMBER: builtins.int
     run_id: builtins.int
-    fab_id: typing.Text
-    fab_version: typing.Text
+    fab_id: builtins.str
+    fab_version: builtins.str
+    fab_hash: builtins.str
+    pending_at: builtins.str
+    starting_at: builtins.str
+    running_at: builtins.str
+    finished_at: builtins.str
+    flwr_aid: builtins.str
+    federation: builtins.str
     @property
-    def override_config(self) -> google.protobuf.internal.containers.MessageMap[typing.Text, flwr.proto.transport_pb2.Scalar]: ...
-    fab_hash: typing.Text
-    pending_at: typing.Text
-    starting_at: typing.Text
-    running_at: typing.Text
-    finished_at: typing.Text
+    def override_config(self) -> google.protobuf.internal.containers.MessageMap[builtins.str, flwr.proto.transport_pb2.Scalar]: ...
     @property
     def status(self) -> global___RunStatus: ...
-    flwr_aid: typing.Text
-    def __init__(self,
+    def __init__(
+        self,
         *,
         run_id: builtins.int = ...,
-        fab_id: typing.Text = ...,
-        fab_version: typing.Text = ...,
-        override_config: typing.Optional[typing.Mapping[typing.Text, flwr.proto.transport_pb2.Scalar]] = ...,
-        fab_hash: typing.Text = ...,
-        pending_at: typing.Text = ...,
-        starting_at: typing.Text = ...,
-        running_at: typing.Text = ...,
-        finished_at: typing.Text = ...,
-        status: typing.Optional[global___RunStatus] = ...,
-        flwr_aid: typing.Text = ...,
-        ) -> None: ...
-    def HasField(self, field_name: typing_extensions.Literal["status",b"status"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["fab_hash",b"fab_hash","fab_id",b"fab_id","fab_version",b"fab_version","finished_at",b"finished_at","flwr_aid",b"flwr_aid","override_config",b"override_config","pending_at",b"pending_at","run_id",b"run_id","running_at",b"running_at","starting_at",b"starting_at","status",b"status"]) -> None: ...
+        fab_id: builtins.str = ...,
+        fab_version: builtins.str = ...,
+        override_config: collections.abc.Mapping[builtins.str, flwr.proto.transport_pb2.Scalar] | None = ...,
+        fab_hash: builtins.str = ...,
+        pending_at: builtins.str = ...,
+        starting_at: builtins.str = ...,
+        running_at: builtins.str = ...,
+        finished_at: builtins.str = ...,
+        status: global___RunStatus | None = ...,
+        flwr_aid: builtins.str = ...,
+        federation: builtins.str = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing.Literal["status", b"status"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["fab_hash", b"fab_hash", "fab_id", b"fab_id", "fab_version", b"fab_version", "federation", b"federation", "finished_at", b"finished_at", "flwr_aid", b"flwr_aid", "override_config", b"override_config", "pending_at", b"pending_at", "run_id", b"run_id", "running_at", b"running_at", "starting_at", b"starting_at", "status", b"status"]) -> None: ...
+
 global___Run = Run
 
+@typing.final
 class RunStatus(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
     STATUS_FIELD_NUMBER: builtins.int
     SUB_STATUS_FIELD_NUMBER: builtins.int
     DETAILS_FIELD_NUMBER: builtins.int
-    status: typing.Text
+    status: builtins.str
     """"starting", "running", "finished" """
-
-    sub_status: typing.Text
+    sub_status: builtins.str
     """"completed", "failed", "stopped" or "" (non-finished)"""
-
-    details: typing.Text
+    details: builtins.str
     """failure details"""
-
-    def __init__(self,
+    def __init__(
+        self,
         *,
-        status: typing.Text = ...,
-        sub_status: typing.Text = ...,
-        details: typing.Text = ...,
-        ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["details",b"details","status",b"status","sub_status",b"sub_status"]) -> None: ...
+        status: builtins.str = ...,
+        sub_status: builtins.str = ...,
+        details: builtins.str = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing.Literal["details", b"details", "status", b"status", "sub_status", b"sub_status"]) -> None: ...
+
 global___RunStatus = RunStatus
 
+@typing.final
 class GetRunRequest(google.protobuf.message.Message):
     """GetRun"""
+
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
     NODE_FIELD_NUMBER: builtins.int
     RUN_ID_FIELD_NUMBER: builtins.int
+    run_id: builtins.int
     @property
     def node(self) -> flwr.proto.node_pb2.Node: ...
-    run_id: builtins.int
-    def __init__(self,
+    def __init__(
+        self,
         *,
-        node: typing.Optional[flwr.proto.node_pb2.Node] = ...,
+        node: flwr.proto.node_pb2.Node | None = ...,
         run_id: builtins.int = ...,
-        ) -> None: ...
-    def HasField(self, field_name: typing_extensions.Literal["node",b"node"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["node",b"node","run_id",b"run_id"]) -> None: ...
+    ) -> None: ...
+    def HasField(self, field_name: typing.Literal["node", b"node"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["node", b"node", "run_id", b"run_id"]) -> None: ...
+
 global___GetRunRequest = GetRunRequest
 
+@typing.final
 class GetRunResponse(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
     RUN_FIELD_NUMBER: builtins.int
     @property
     def run(self) -> global___Run: ...
-    def __init__(self,
+    def __init__(
+        self,
         *,
-        run: typing.Optional[global___Run] = ...,
-        ) -> None: ...
-    def HasField(self, field_name: typing_extensions.Literal["run",b"run"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["run",b"run"]) -> None: ...
+        run: global___Run | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing.Literal["run", b"run"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["run", b"run"]) -> None: ...
+
 global___GetRunResponse = GetRunResponse
 
+@typing.final
 class UpdateRunStatusRequest(google.protobuf.message.Message):
     """UpdateRunStatus"""
+
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
     RUN_ID_FIELD_NUMBER: builtins.int
     RUN_STATUS_FIELD_NUMBER: builtins.int
     run_id: builtins.int
     @property
     def run_status(self) -> global___RunStatus: ...
-    def __init__(self,
+    def __init__(
+        self,
         *,
         run_id: builtins.int = ...,
-        run_status: typing.Optional[global___RunStatus] = ...,
-        ) -> None: ...
-    def HasField(self, field_name: typing_extensions.Literal["run_status",b"run_status"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["run_id",b"run_id","run_status",b"run_status"]) -> None: ...
+        run_status: global___RunStatus | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing.Literal["run_status", b"run_status"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["run_id", b"run_id", "run_status", b"run_status"]) -> None: ...
+
 global___UpdateRunStatusRequest = UpdateRunStatusRequest
 
+@typing.final
 class UpdateRunStatusResponse(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
-    def __init__(self,
-        ) -> None: ...
+
+    def __init__(
+        self,
+    ) -> None: ...
+
 global___UpdateRunStatusResponse = UpdateRunStatusResponse
 
+@typing.final
 class GetRunStatusRequest(google.protobuf.message.Message):
     """GetRunStatus"""
+
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
     NODE_FIELD_NUMBER: builtins.int
     RUN_IDS_FIELD_NUMBER: builtins.int
     @property
     def node(self) -> flwr.proto.node_pb2.Node: ...
     @property
     def run_ids(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.int]: ...
-    def __init__(self,
+    def __init__(
+        self,
         *,
-        node: typing.Optional[flwr.proto.node_pb2.Node] = ...,
-        run_ids: typing.Optional[typing.Iterable[builtins.int]] = ...,
-        ) -> None: ...
-    def HasField(self, field_name: typing_extensions.Literal["node",b"node"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["node",b"node","run_ids",b"run_ids"]) -> None: ...
+        node: flwr.proto.node_pb2.Node | None = ...,
+        run_ids: collections.abc.Iterable[builtins.int] | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing.Literal["node", b"node"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["node", b"node", "run_ids", b"run_ids"]) -> None: ...
+
 global___GetRunStatusRequest = GetRunStatusRequest
 
+@typing.final
 class GetRunStatusResponse(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    @typing.final
     class RunStatusDictEntry(google.protobuf.message.Message):
         DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
         KEY_FIELD_NUMBER: builtins.int
         VALUE_FIELD_NUMBER: builtins.int
         key: builtins.int
         @property
         def value(self) -> global___RunStatus: ...
-        def __init__(self,
+        def __init__(
+            self,
             *,
             key: builtins.int = ...,
-            value: typing.Optional[global___RunStatus] = ...,
-            ) -> 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: ...
+            value: global___RunStatus | None = ...,
+        ) -> None: ...
+        def HasField(self, field_name: typing.Literal["value", b"value"]) -> builtins.bool: ...
+        def ClearField(self, field_name: typing.Literal["key", b"key", "value", b"value"]) -> None: ...
 
     RUN_STATUS_DICT_FIELD_NUMBER: builtins.int
     @property
     def run_status_dict(self) -> google.protobuf.internal.containers.MessageMap[builtins.int, global___RunStatus]: ...
-    def __init__(self,
+    def __init__(
+        self,
         *,
-        run_status_dict: typing.Optional[typing.Mapping[builtins.int, global___RunStatus]] = ...,
-        ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["run_status_dict",b"run_status_dict"]) -> None: ...
+        run_status_dict: collections.abc.Mapping[builtins.int, global___RunStatus] | None = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing.Literal["run_status_dict", b"run_status_dict"]) -> None: ...
+
 global___GetRunStatusResponse = GetRunStatusResponse
 
+@typing.final
 class GetFederationOptionsRequest(google.protobuf.message.Message):
     """Get Federation Options associated with run"""
+
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
     RUN_ID_FIELD_NUMBER: builtins.int
     run_id: builtins.int
-    def __init__(self,
+    def __init__(
+        self,
         *,
         run_id: builtins.int = ...,
-        ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["run_id",b"run_id"]) -> None: ...
+    ) -> None: ...
+    def ClearField(self, field_name: typing.Literal["run_id", b"run_id"]) -> None: ...
+
 global___GetFederationOptionsRequest = GetFederationOptionsRequest
 
+@typing.final
 class GetFederationOptionsResponse(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
     FEDERATION_OPTIONS_FIELD_NUMBER: builtins.int
     @property
     def federation_options(self) -> flwr.proto.recorddict_pb2.ConfigRecord: ...
-    def __init__(self,
+    def __init__(
+        self,
         *,
-        federation_options: typing.Optional[flwr.proto.recorddict_pb2.ConfigRecord] = ...,
-        ) -> None: ...
-    def HasField(self, field_name: typing_extensions.Literal["federation_options",b"federation_options"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["federation_options",b"federation_options"]) -> None: ...
+        federation_options: flwr.proto.recorddict_pb2.ConfigRecord | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing.Literal["federation_options", b"federation_options"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["federation_options", b"federation_options"]) -> None: ...
+
 global___GetFederationOptionsResponse = GetFederationOptionsResponse

flwr/proto/run_pb2_grpc.py

@@ -1,4 +1,24 @@
 # Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
 """Client and server classes corresponding to protobuf-defined services."""
 import grpc
+import warnings
 
+
+GRPC_GENERATED_VERSION = '1.70.0'
+GRPC_VERSION = grpc.__version__
+_version_not_supported = False
+
+try:
+    from grpc._utilities import first_version_is_lower
+    _version_not_supported = first_version_is_lower(GRPC_VERSION, GRPC_GENERATED_VERSION)
+except ImportError:
+    _version_not_supported = True
+
+if _version_not_supported:
+    raise RuntimeError(
+        f'The grpc package installed is at version {GRPC_VERSION},'
+        + f' but the generated code in flwr/proto/run_pb2_grpc.py depends on'
+        + f' grpcio>={GRPC_GENERATED_VERSION}.'
+        + f' Please upgrade your grpc module to grpcio>={GRPC_GENERATED_VERSION}'
+        + f' or downgrade your generated code using grpcio-tools<={GRPC_VERSION}.'
+    )

flwr/proto/run_pb2_grpc.pyi

@@ -1,4 +1,31 @@
 """
 @generated by mypy-protobuf.  Do not edit manually!
 isort:skip_file
+Copyright 2024 Flower Labs GmbH. All Rights Reserved.
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+==============================================================================
 """
+
+import abc
+import collections.abc
+import grpc
+import grpc.aio
+import typing
+
+_T = typing.TypeVar("_T")
+
+class _MaybeAsyncIterator(collections.abc.AsyncIterator[_T], collections.abc.Iterator[_T], metaclass=abc.ABCMeta): ...
+
+class _ServicerContext(grpc.ServicerContext, grpc.aio.ServicerContext):  # type: ignore[misc, type-arg]
+    ...

flwr/proto/serverappio_pb2.py

@@ -1,12 +1,22 @@
 # -*- coding: utf-8 -*-
 # Generated by the protocol buffer compiler.  DO NOT EDIT!
+# NO CHECKED-IN PROTOBUF GENCODE
 # source: flwr/proto/serverappio.proto
-# Protobuf Python Version: 4.25.1
+# Protobuf Python Version: 5.29.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 runtime_version as _runtime_version
 from google.protobuf import symbol_database as _symbol_database
 from google.protobuf.internal import builder as _builder
+_runtime_version.ValidateProtobufRuntimeVersion(
+    _runtime_version.Domain.PUBLIC,
+    5,
+    29,
+    0,
+    '',
+    'flwr/proto/serverappio.proto'
+)
 # @@protoc_insertion_point(imports)
 
 _sym_db = _symbol_database.Default()
@@ -26,8 +36,8 @@ DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1c\x66lwr/proto/
 _globals = globals()
 _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
 _builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'flwr.proto.serverappio_pb2', _globals)
-if _descriptor._USE_C_DESCRIPTORS == False:
-  DESCRIPTOR._options = None
+if not _descriptor._USE_C_DESCRIPTORS:
+  DESCRIPTOR._loaded_options = None
   _globals['_GETNODESREQUEST']._serialized_start=211
   _globals['_GETNODESREQUEST']._serialized_end=244
   _globals['_GETNODESRESPONSE']._serialized_start=246

flwr/proto/serverappio_pb2.pyi

@@ -1,37 +1,61 @@
 """
 @generated by mypy-protobuf.  Do not edit manually!
 isort:skip_file
+Copyright 2022 Flower Labs GmbH. All Rights Reserved.
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+    http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+==============================================================================
 """
+
 import builtins
+import collections.abc
 import flwr.proto.node_pb2
 import google.protobuf.descriptor
 import google.protobuf.internal.containers
 import google.protobuf.message
 import typing
-import typing_extensions
 
 DESCRIPTOR: google.protobuf.descriptor.FileDescriptor
 
+@typing.final
 class GetNodesRequest(google.protobuf.message.Message):
     """GetNodes messages"""
+
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
     RUN_ID_FIELD_NUMBER: builtins.int
     run_id: builtins.int
-    def __init__(self,
+    def __init__(
+        self,
         *,
         run_id: builtins.int = ...,
-        ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["run_id",b"run_id"]) -> None: ...
+    ) -> None: ...
+    def ClearField(self, field_name: typing.Literal["run_id", b"run_id"]) -> None: ...
+
 global___GetNodesRequest = GetNodesRequest
 
+@typing.final
 class GetNodesResponse(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
     NODES_FIELD_NUMBER: builtins.int
     @property
     def nodes(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[flwr.proto.node_pb2.Node]: ...
-    def __init__(self,
+    def __init__(
+        self,
         *,
-        nodes: typing.Optional[typing.Iterable[flwr.proto.node_pb2.Node]] = ...,
-        ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["nodes",b"nodes"]) -> None: ...
+        nodes: collections.abc.Iterable[flwr.proto.node_pb2.Node] | None = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing.Literal["nodes", b"nodes"]) -> None: ...
+
 global___GetNodesResponse = GetNodesResponse