flwr 1.23.0__py3-none-any.whl → 1.25.0__py3-none-any.whl

flwr/proto/node_pb2_grpc.pyi

@@ -1,4 +1,31 @@
 """
 @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 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/recorddict_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/recorddict.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/recorddict.proto'
+)
 # @@protoc_insertion_point(imports)
 
 _sym_db = _symbol_database.Default()
@@ -19,8 +29,8 @@ DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x1b\x66lwr/proto/
 _globals = globals()
 _builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
 _builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'flwr.proto.recorddict_pb2', _globals)
-if _descriptor._USE_C_DESCRIPTORS == False:
-  DESCRIPTOR._options = None
+if not _descriptor._USE_C_DESCRIPTORS:
+  DESCRIPTOR._loaded_options = None
   _globals['_DOUBLELIST']._serialized_start=43
   _globals['_DOUBLELIST']._serialized_end=69
   _globals['_SINTLIST']._serialized_start=71

flwr/proto/recorddict_pb2.pyi

@@ -1,111 +1,156 @@
 """
 @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 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 DoubleList(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
     VALS_FIELD_NUMBER: builtins.int
     @property
     def vals(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.float]: ...
-    def __init__(self,
+    def __init__(
+        self,
         *,
-        vals: typing.Optional[typing.Iterable[builtins.float]] = ...,
-        ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["vals",b"vals"]) -> None: ...
+        vals: collections.abc.Iterable[builtins.float] | None = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing.Literal["vals", b"vals"]) -> None: ...
+
 global___DoubleList = DoubleList
 
+@typing.final
 class SintList(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
     VALS_FIELD_NUMBER: builtins.int
     @property
     def vals(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.int]: ...
-    def __init__(self,
+    def __init__(
+        self,
         *,
-        vals: typing.Optional[typing.Iterable[builtins.int]] = ...,
-        ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["vals",b"vals"]) -> None: ...
+        vals: collections.abc.Iterable[builtins.int] | None = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing.Literal["vals", b"vals"]) -> None: ...
+
 global___SintList = SintList
 
+@typing.final
 class UintList(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
     VALS_FIELD_NUMBER: builtins.int
     @property
     def vals(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.int]: ...
-    def __init__(self,
+    def __init__(
+        self,
         *,
-        vals: typing.Optional[typing.Iterable[builtins.int]] = ...,
-        ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["vals",b"vals"]) -> None: ...
+        vals: collections.abc.Iterable[builtins.int] | None = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing.Literal["vals", b"vals"]) -> None: ...
+
 global___UintList = UintList
 
+@typing.final
 class BoolList(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
     VALS_FIELD_NUMBER: builtins.int
     @property
     def vals(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.bool]: ...
-    def __init__(self,
+    def __init__(
+        self,
         *,
-        vals: typing.Optional[typing.Iterable[builtins.bool]] = ...,
-        ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["vals",b"vals"]) -> None: ...
+        vals: collections.abc.Iterable[builtins.bool] | None = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing.Literal["vals", b"vals"]) -> None: ...
+
 global___BoolList = BoolList
 
+@typing.final
 class StringList(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
     VALS_FIELD_NUMBER: builtins.int
     @property
-    def vals(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[typing.Text]: ...
-    def __init__(self,
+    def vals(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.str]: ...
+    def __init__(
+        self,
         *,
-        vals: typing.Optional[typing.Iterable[typing.Text]] = ...,
-        ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["vals",b"vals"]) -> None: ...
+        vals: collections.abc.Iterable[builtins.str] | None = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing.Literal["vals", b"vals"]) -> None: ...
+
 global___StringList = StringList
 
+@typing.final
 class BytesList(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
     VALS_FIELD_NUMBER: builtins.int
     @property
     def vals(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.bytes]: ...
-    def __init__(self,
+    def __init__(
+        self,
         *,
-        vals: typing.Optional[typing.Iterable[builtins.bytes]] = ...,
-        ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["vals",b"vals"]) -> None: ...
+        vals: collections.abc.Iterable[builtins.bytes] | None = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing.Literal["vals", b"vals"]) -> None: ...
+
 global___BytesList = BytesList
 
+@typing.final
 class Array(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
     DTYPE_FIELD_NUMBER: builtins.int
     SHAPE_FIELD_NUMBER: builtins.int
     STYPE_FIELD_NUMBER: builtins.int
     DATA_FIELD_NUMBER: builtins.int
-    dtype: typing.Text
+    dtype: builtins.str
+    stype: builtins.str
+    data: builtins.bytes
     @property
     def shape(self) -> google.protobuf.internal.containers.RepeatedScalarFieldContainer[builtins.int]: ...
-    stype: typing.Text
-    data: builtins.bytes
-    def __init__(self,
+    def __init__(
+        self,
         *,
-        dtype: typing.Text = ...,
-        shape: typing.Optional[typing.Iterable[builtins.int]] = ...,
-        stype: typing.Text = ...,
+        dtype: builtins.str = ...,
+        shape: collections.abc.Iterable[builtins.int] | None = ...,
+        stype: builtins.str = ...,
         data: builtins.bytes = ...,
-        ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["data",b"data","dtype",b"dtype","shape",b"shape","stype",b"stype"]) -> None: ...
+    ) -> None: ...
+    def ClearField(self, field_name: typing.Literal["data", b"data", "dtype", b"dtype", "shape", b"shape", "stype", b"stype"]) -> None: ...
+
 global___Array = Array
 
+@typing.final
 class MetricRecordValue(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
     DOUBLE_FIELD_NUMBER: builtins.int
     SINT64_FIELD_NUMBER: builtins.int
     UINT64_FIELD_NUMBER: builtins.int
@@ -114,33 +159,36 @@ class MetricRecordValue(google.protobuf.message.Message):
     UINT_LIST_FIELD_NUMBER: builtins.int
     double: builtins.float
     """Single element"""
-
     sint64: builtins.int
     uint64: builtins.int
     @property
     def double_list(self) -> global___DoubleList:
         """List types"""
-        pass
+
     @property
     def sint_list(self) -> global___SintList: ...
     @property
     def uint_list(self) -> global___UintList: ...
-    def __init__(self,
+    def __init__(
+        self,
         *,
         double: builtins.float = ...,
         sint64: builtins.int = ...,
         uint64: builtins.int = ...,
-        double_list: typing.Optional[global___DoubleList] = ...,
-        sint_list: typing.Optional[global___SintList] = ...,
-        uint_list: typing.Optional[global___UintList] = ...,
-        ) -> None: ...
-    def HasField(self, field_name: typing_extensions.Literal["double",b"double","double_list",b"double_list","sint64",b"sint64","sint_list",b"sint_list","uint64",b"uint64","uint_list",b"uint_list","value",b"value"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["double",b"double","double_list",b"double_list","sint64",b"sint64","sint_list",b"sint_list","uint64",b"uint64","uint_list",b"uint_list","value",b"value"]) -> None: ...
-    def WhichOneof(self, oneof_group: typing_extensions.Literal["value",b"value"]) -> typing.Optional[typing_extensions.Literal["double","sint64","uint64","double_list","sint_list","uint_list"]]: ...
+        double_list: global___DoubleList | None = ...,
+        sint_list: global___SintList | None = ...,
+        uint_list: global___UintList | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing.Literal["double", b"double", "double_list", b"double_list", "sint64", b"sint64", "sint_list", b"sint_list", "uint64", b"uint64", "uint_list", b"uint_list", "value", b"value"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["double", b"double", "double_list", b"double_list", "sint64", b"sint64", "sint_list", b"sint_list", "uint64", b"uint64", "uint_list", b"uint_list", "value", b"value"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing.Literal["value", b"value"]) -> typing.Literal["double", "sint64", "uint64", "double_list", "sint_list", "uint_list"] | None: ...
+
 global___MetricRecordValue = MetricRecordValue
 
+@typing.final
 class ConfigRecordValue(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
     DOUBLE_FIELD_NUMBER: builtins.int
     SINT64_FIELD_NUMBER: builtins.int
     UINT64_FIELD_NUMBER: builtins.int
@@ -155,16 +203,15 @@ class ConfigRecordValue(google.protobuf.message.Message):
     BYTES_LIST_FIELD_NUMBER: builtins.int
     double: builtins.float
     """Single element"""
-
     sint64: builtins.int
     uint64: builtins.int
     bool: builtins.bool
-    string: typing.Text
+    string: builtins.str
     bytes: builtins.bytes
     @property
     def double_list(self) -> global___DoubleList:
         """List types"""
-        pass
+
     @property
     def sint_list(self) -> global___SintList: ...
     @property
@@ -175,139 +222,169 @@ class ConfigRecordValue(google.protobuf.message.Message):
     def string_list(self) -> global___StringList: ...
     @property
     def bytes_list(self) -> global___BytesList: ...
-    def __init__(self,
+    def __init__(
+        self,
         *,
         double: builtins.float = ...,
         sint64: builtins.int = ...,
         uint64: builtins.int = ...,
         bool: builtins.bool = ...,
-        string: typing.Text = ...,
+        string: builtins.str = ...,
         bytes: builtins.bytes = ...,
-        double_list: typing.Optional[global___DoubleList] = ...,
-        sint_list: typing.Optional[global___SintList] = ...,
-        uint_list: typing.Optional[global___UintList] = ...,
-        bool_list: typing.Optional[global___BoolList] = ...,
-        string_list: typing.Optional[global___StringList] = ...,
-        bytes_list: typing.Optional[global___BytesList] = ...,
-        ) -> None: ...
-    def HasField(self, field_name: typing_extensions.Literal["bool",b"bool","bool_list",b"bool_list","bytes",b"bytes","bytes_list",b"bytes_list","double",b"double","double_list",b"double_list","sint64",b"sint64","sint_list",b"sint_list","string",b"string","string_list",b"string_list","uint64",b"uint64","uint_list",b"uint_list","value",b"value"]) -> builtins.bool: ...
-    def ClearField(self, field_name: typing_extensions.Literal["bool",b"bool","bool_list",b"bool_list","bytes",b"bytes","bytes_list",b"bytes_list","double",b"double","double_list",b"double_list","sint64",b"sint64","sint_list",b"sint_list","string",b"string","string_list",b"string_list","uint64",b"uint64","uint_list",b"uint_list","value",b"value"]) -> None: ...
-    def WhichOneof(self, oneof_group: typing_extensions.Literal["value",b"value"]) -> typing.Optional[typing_extensions.Literal["double","sint64","uint64","bool","string","bytes","double_list","sint_list","uint_list","bool_list","string_list","bytes_list"]]: ...
+        double_list: global___DoubleList | None = ...,
+        sint_list: global___SintList | None = ...,
+        uint_list: global___UintList | None = ...,
+        bool_list: global___BoolList | None = ...,
+        string_list: global___StringList | None = ...,
+        bytes_list: global___BytesList | None = ...,
+    ) -> None: ...
+    def HasField(self, field_name: typing.Literal["bool", b"bool", "bool_list", b"bool_list", "bytes", b"bytes", "bytes_list", b"bytes_list", "double", b"double", "double_list", b"double_list", "sint64", b"sint64", "sint_list", b"sint_list", "string", b"string", "string_list", b"string_list", "uint64", b"uint64", "uint_list", b"uint_list", "value", b"value"]) -> builtins.bool: ...
+    def ClearField(self, field_name: typing.Literal["bool", b"bool", "bool_list", b"bool_list", "bytes", b"bytes", "bytes_list", b"bytes_list", "double", b"double", "double_list", b"double_list", "sint64", b"sint64", "sint_list", b"sint_list", "string", b"string", "string_list", b"string_list", "uint64", b"uint64", "uint_list", b"uint_list", "value", b"value"]) -> None: ...
+    def WhichOneof(self, oneof_group: typing.Literal["value", b"value"]) -> typing.Literal["double", "sint64", "uint64", "bool", "string", "bytes", "double_list", "sint_list", "uint_list", "bool_list", "string_list", "bytes_list"] | None: ...
+
 global___ConfigRecordValue = ConfigRecordValue
 
+@typing.final
 class ArrayRecord(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    @typing.final
     class Item(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) -> global___Array: ...
-        def __init__(self,
+        def __init__(
+            self,
             *,
-            key: typing.Text = ...,
-            value: typing.Optional[global___Array] = ...,
-            ) -> 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: global___Array | 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: ...
 
     ITEMS_FIELD_NUMBER: builtins.int
     @property
     def items(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___ArrayRecord.Item]: ...
-    def __init__(self,
+    def __init__(
+        self,
         *,
-        items: typing.Optional[typing.Iterable[global___ArrayRecord.Item]] = ...,
-        ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["items",b"items"]) -> None: ...
+        items: collections.abc.Iterable[global___ArrayRecord.Item] | None = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing.Literal["items", b"items"]) -> None: ...
+
 global___ArrayRecord = ArrayRecord
 
+@typing.final
 class MetricRecord(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    @typing.final
     class Item(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) -> global___MetricRecordValue: ...
-        def __init__(self,
+        def __init__(
+            self,
             *,
-            key: typing.Text = ...,
-            value: typing.Optional[global___MetricRecordValue] = ...,
-            ) -> 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: global___MetricRecordValue | 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: ...
 
     ITEMS_FIELD_NUMBER: builtins.int
     @property
     def items(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___MetricRecord.Item]: ...
-    def __init__(self,
+    def __init__(
+        self,
         *,
-        items: typing.Optional[typing.Iterable[global___MetricRecord.Item]] = ...,
-        ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["items",b"items"]) -> None: ...
+        items: collections.abc.Iterable[global___MetricRecord.Item] | None = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing.Literal["items", b"items"]) -> None: ...
+
 global___MetricRecord = MetricRecord
 
+@typing.final
 class ConfigRecord(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    @typing.final
     class Item(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) -> global___ConfigRecordValue: ...
-        def __init__(self,
+        def __init__(
+            self,
             *,
-            key: typing.Text = ...,
-            value: typing.Optional[global___ConfigRecordValue] = ...,
-            ) -> 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: global___ConfigRecordValue | 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: ...
 
     ITEMS_FIELD_NUMBER: builtins.int
     @property
     def items(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___ConfigRecord.Item]: ...
-    def __init__(self,
+    def __init__(
+        self,
         *,
-        items: typing.Optional[typing.Iterable[global___ConfigRecord.Item]] = ...,
-        ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["items",b"items"]) -> None: ...
+        items: collections.abc.Iterable[global___ConfigRecord.Item] | None = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing.Literal["items", b"items"]) -> None: ...
+
 global___ConfigRecord = ConfigRecord
 
+@typing.final
 class RecordDict(google.protobuf.message.Message):
     DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
+    @typing.final
     class Item(google.protobuf.message.Message):
         DESCRIPTOR: google.protobuf.descriptor.Descriptor
+
         KEY_FIELD_NUMBER: builtins.int
         ARRAY_RECORD_FIELD_NUMBER: builtins.int
         METRIC_RECORD_FIELD_NUMBER: builtins.int
         CONFIG_RECORD_FIELD_NUMBER: builtins.int
-        key: typing.Text
+        key: builtins.str
         @property
         def array_record(self) -> global___ArrayRecord: ...
         @property
         def metric_record(self) -> global___MetricRecord: ...
         @property
         def config_record(self) -> global___ConfigRecord: ...
-        def __init__(self,
+        def __init__(
+            self,
             *,
-            key: typing.Text = ...,
-            array_record: typing.Optional[global___ArrayRecord] = ...,
-            metric_record: typing.Optional[global___MetricRecord] = ...,
-            config_record: typing.Optional[global___ConfigRecord] = ...,
-            ) -> None: ...
-        def HasField(self, field_name: typing_extensions.Literal["array_record",b"array_record","config_record",b"config_record","metric_record",b"metric_record","value",b"value"]) -> builtins.bool: ...
-        def ClearField(self, field_name: typing_extensions.Literal["array_record",b"array_record","config_record",b"config_record","key",b"key","metric_record",b"metric_record","value",b"value"]) -> None: ...
-        def WhichOneof(self, oneof_group: typing_extensions.Literal["value",b"value"]) -> typing.Optional[typing_extensions.Literal["array_record","metric_record","config_record"]]: ...
+            key: builtins.str = ...,
+            array_record: global___ArrayRecord | None = ...,
+            metric_record: global___MetricRecord | None = ...,
+            config_record: global___ConfigRecord | None = ...,
+        ) -> None: ...
+        def HasField(self, field_name: typing.Literal["array_record", b"array_record", "config_record", b"config_record", "metric_record", b"metric_record", "value", b"value"]) -> builtins.bool: ...
+        def ClearField(self, field_name: typing.Literal["array_record", b"array_record", "config_record", b"config_record", "key", b"key", "metric_record", b"metric_record", "value", b"value"]) -> None: ...
+        def WhichOneof(self, oneof_group: typing.Literal["value", b"value"]) -> typing.Literal["array_record", "metric_record", "config_record"] | None: ...
 
     ITEMS_FIELD_NUMBER: builtins.int
     @property
     def items(self) -> google.protobuf.internal.containers.RepeatedCompositeFieldContainer[global___RecordDict.Item]: ...
-    def __init__(self,
+    def __init__(
+        self,
         *,
-        items: typing.Optional[typing.Iterable[global___RecordDict.Item]] = ...,
-        ) -> None: ...
-    def ClearField(self, field_name: typing_extensions.Literal["items",b"items"]) -> None: ...
+        items: collections.abc.Iterable[global___RecordDict.Item] | None = ...,
+    ) -> None: ...
+    def ClearField(self, field_name: typing.Literal["items", b"items"]) -> None: ...
+
 global___RecordDict = RecordDict

flwr/proto/recorddict_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/recorddict_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/recorddict_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]
+    ...