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

flwr/proto/simulationio_pb2_grpc.py

@@ -1,12 +1,32 @@
 # Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
 """Client and server classes corresponding to protobuf-defined services."""
 import grpc
+import warnings
 
 from flwr.proto import appio_pb2 as flwr_dot_proto_dot_appio__pb2
 from flwr.proto import heartbeat_pb2 as flwr_dot_proto_dot_heartbeat__pb2
 from flwr.proto import log_pb2 as flwr_dot_proto_dot_log__pb2
 from flwr.proto import run_pb2 as flwr_dot_proto_dot_run__pb2
 
+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/simulationio_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}.'
+    )
+
 
 class SimulationIoStub(object):
     """Missing associated documentation comment in .proto file."""
@@ -21,52 +41,52 @@ class SimulationIoStub(object):
                 '/flwr.proto.SimulationIo/ListAppsToLaunch',
                 request_serializer=flwr_dot_proto_dot_appio__pb2.ListAppsToLaunchRequest.SerializeToString,
                 response_deserializer=flwr_dot_proto_dot_appio__pb2.ListAppsToLaunchResponse.FromString,
-                )
+                _registered_method=True)
         self.RequestToken = channel.unary_unary(
                 '/flwr.proto.SimulationIo/RequestToken',
                 request_serializer=flwr_dot_proto_dot_appio__pb2.RequestTokenRequest.SerializeToString,
                 response_deserializer=flwr_dot_proto_dot_appio__pb2.RequestTokenResponse.FromString,
-                )
+                _registered_method=True)
         self.GetRun = channel.unary_unary(
                 '/flwr.proto.SimulationIo/GetRun',
                 request_serializer=flwr_dot_proto_dot_run__pb2.GetRunRequest.SerializeToString,
                 response_deserializer=flwr_dot_proto_dot_run__pb2.GetRunResponse.FromString,
-                )
+                _registered_method=True)
         self.PullAppInputs = channel.unary_unary(
                 '/flwr.proto.SimulationIo/PullAppInputs',
                 request_serializer=flwr_dot_proto_dot_appio__pb2.PullAppInputsRequest.SerializeToString,
                 response_deserializer=flwr_dot_proto_dot_appio__pb2.PullAppInputsResponse.FromString,
-                )
+                _registered_method=True)
         self.PushAppOutputs = channel.unary_unary(
                 '/flwr.proto.SimulationIo/PushAppOutputs',
                 request_serializer=flwr_dot_proto_dot_appio__pb2.PushAppOutputsRequest.SerializeToString,
                 response_deserializer=flwr_dot_proto_dot_appio__pb2.PushAppOutputsResponse.FromString,
-                )
+                _registered_method=True)
         self.UpdateRunStatus = channel.unary_unary(
                 '/flwr.proto.SimulationIo/UpdateRunStatus',
                 request_serializer=flwr_dot_proto_dot_run__pb2.UpdateRunStatusRequest.SerializeToString,
                 response_deserializer=flwr_dot_proto_dot_run__pb2.UpdateRunStatusResponse.FromString,
-                )
+                _registered_method=True)
         self.PushLogs = channel.unary_unary(
                 '/flwr.proto.SimulationIo/PushLogs',
                 request_serializer=flwr_dot_proto_dot_log__pb2.PushLogsRequest.SerializeToString,
                 response_deserializer=flwr_dot_proto_dot_log__pb2.PushLogsResponse.FromString,
-                )
+                _registered_method=True)
         self.GetFederationOptions = channel.unary_unary(
                 '/flwr.proto.SimulationIo/GetFederationOptions',
                 request_serializer=flwr_dot_proto_dot_run__pb2.GetFederationOptionsRequest.SerializeToString,
                 response_deserializer=flwr_dot_proto_dot_run__pb2.GetFederationOptionsResponse.FromString,
-                )
+                _registered_method=True)
         self.GetRunStatus = channel.unary_unary(
                 '/flwr.proto.SimulationIo/GetRunStatus',
                 request_serializer=flwr_dot_proto_dot_run__pb2.GetRunStatusRequest.SerializeToString,
                 response_deserializer=flwr_dot_proto_dot_run__pb2.GetRunStatusResponse.FromString,
-                )
+                _registered_method=True)
         self.SendAppHeartbeat = channel.unary_unary(
                 '/flwr.proto.SimulationIo/SendAppHeartbeat',
                 request_serializer=flwr_dot_proto_dot_heartbeat__pb2.SendAppHeartbeatRequest.SerializeToString,
                 response_deserializer=flwr_dot_proto_dot_heartbeat__pb2.SendAppHeartbeatResponse.FromString,
-                )
+                _registered_method=True)
 
 
 class SimulationIoServicer(object):
@@ -136,7 +156,7 @@ class SimulationIoServicer(object):
         raise NotImplementedError('Method not implemented!')
 
     def SendAppHeartbeat(self, request, context):
-        """Heartbeat
+        """App heartbeat
         """
         context.set_code(grpc.StatusCode.UNIMPLEMENTED)
         context.set_details('Method not implemented!')
@@ -199,6 +219,7 @@ def add_SimulationIoServicer_to_server(servicer, server):
     generic_handler = grpc.method_handlers_generic_handler(
             'flwr.proto.SimulationIo', rpc_method_handlers)
     server.add_generic_rpc_handlers((generic_handler,))
+    server.add_registered_method_handlers('flwr.proto.SimulationIo', rpc_method_handlers)
 
 
  # This class is part of an EXPERIMENTAL API.
@@ -216,11 +237,21 @@ class SimulationIo(object):
             wait_for_ready=None,
             timeout=None,
             metadata=None):
-        return grpc.experimental.unary_unary(request, target, '/flwr.proto.SimulationIo/ListAppsToLaunch',
+        return grpc.experimental.unary_unary(
+            request,
+            target,
+            '/flwr.proto.SimulationIo/ListAppsToLaunch',
             flwr_dot_proto_dot_appio__pb2.ListAppsToLaunchRequest.SerializeToString,
             flwr_dot_proto_dot_appio__pb2.ListAppsToLaunchResponse.FromString,
-            options, channel_credentials,
-            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)
 
     @staticmethod
     def RequestToken(request,
@@ -233,11 +264,21 @@ class SimulationIo(object):
             wait_for_ready=None,
             timeout=None,
             metadata=None):
-        return grpc.experimental.unary_unary(request, target, '/flwr.proto.SimulationIo/RequestToken',
+        return grpc.experimental.unary_unary(
+            request,
+            target,
+            '/flwr.proto.SimulationIo/RequestToken',
             flwr_dot_proto_dot_appio__pb2.RequestTokenRequest.SerializeToString,
             flwr_dot_proto_dot_appio__pb2.RequestTokenResponse.FromString,
-            options, channel_credentials,
-            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)
 
     @staticmethod
     def GetRun(request,
@@ -250,11 +291,21 @@ class SimulationIo(object):
             wait_for_ready=None,
             timeout=None,
             metadata=None):
-        return grpc.experimental.unary_unary(request, target, '/flwr.proto.SimulationIo/GetRun',
+        return grpc.experimental.unary_unary(
+            request,
+            target,
+            '/flwr.proto.SimulationIo/GetRun',
             flwr_dot_proto_dot_run__pb2.GetRunRequest.SerializeToString,
             flwr_dot_proto_dot_run__pb2.GetRunResponse.FromString,
-            options, channel_credentials,
-            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)
 
     @staticmethod
     def PullAppInputs(request,
@@ -267,11 +318,21 @@ class SimulationIo(object):
             wait_for_ready=None,
             timeout=None,
             metadata=None):
-        return grpc.experimental.unary_unary(request, target, '/flwr.proto.SimulationIo/PullAppInputs',
+        return grpc.experimental.unary_unary(
+            request,
+            target,
+            '/flwr.proto.SimulationIo/PullAppInputs',
             flwr_dot_proto_dot_appio__pb2.PullAppInputsRequest.SerializeToString,
             flwr_dot_proto_dot_appio__pb2.PullAppInputsResponse.FromString,
-            options, channel_credentials,
-            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)
 
     @staticmethod
     def PushAppOutputs(request,
@@ -284,11 +345,21 @@ class SimulationIo(object):
             wait_for_ready=None,
             timeout=None,
             metadata=None):
-        return grpc.experimental.unary_unary(request, target, '/flwr.proto.SimulationIo/PushAppOutputs',
+        return grpc.experimental.unary_unary(
+            request,
+            target,
+            '/flwr.proto.SimulationIo/PushAppOutputs',
             flwr_dot_proto_dot_appio__pb2.PushAppOutputsRequest.SerializeToString,
             flwr_dot_proto_dot_appio__pb2.PushAppOutputsResponse.FromString,
-            options, channel_credentials,
-            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)
 
     @staticmethod
     def UpdateRunStatus(request,
@@ -301,11 +372,21 @@ class SimulationIo(object):
             wait_for_ready=None,
             timeout=None,
             metadata=None):
-        return grpc.experimental.unary_unary(request, target, '/flwr.proto.SimulationIo/UpdateRunStatus',
+        return grpc.experimental.unary_unary(
+            request,
+            target,
+            '/flwr.proto.SimulationIo/UpdateRunStatus',
             flwr_dot_proto_dot_run__pb2.UpdateRunStatusRequest.SerializeToString,
             flwr_dot_proto_dot_run__pb2.UpdateRunStatusResponse.FromString,
-            options, channel_credentials,
-            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)
 
     @staticmethod
     def PushLogs(request,
@@ -318,11 +399,21 @@ class SimulationIo(object):
             wait_for_ready=None,
             timeout=None,
             metadata=None):
-        return grpc.experimental.unary_unary(request, target, '/flwr.proto.SimulationIo/PushLogs',
+        return grpc.experimental.unary_unary(
+            request,
+            target,
+            '/flwr.proto.SimulationIo/PushLogs',
             flwr_dot_proto_dot_log__pb2.PushLogsRequest.SerializeToString,
             flwr_dot_proto_dot_log__pb2.PushLogsResponse.FromString,
-            options, channel_credentials,
-            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)
 
     @staticmethod
     def GetFederationOptions(request,
@@ -335,11 +426,21 @@ class SimulationIo(object):
             wait_for_ready=None,
             timeout=None,
             metadata=None):
-        return grpc.experimental.unary_unary(request, target, '/flwr.proto.SimulationIo/GetFederationOptions',
+        return grpc.experimental.unary_unary(
+            request,
+            target,
+            '/flwr.proto.SimulationIo/GetFederationOptions',
             flwr_dot_proto_dot_run__pb2.GetFederationOptionsRequest.SerializeToString,
             flwr_dot_proto_dot_run__pb2.GetFederationOptionsResponse.FromString,
-            options, channel_credentials,
-            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)
 
     @staticmethod
     def GetRunStatus(request,
@@ -352,11 +453,21 @@ class SimulationIo(object):
             wait_for_ready=None,
             timeout=None,
             metadata=None):
-        return grpc.experimental.unary_unary(request, target, '/flwr.proto.SimulationIo/GetRunStatus',
+        return grpc.experimental.unary_unary(
+            request,
+            target,
+            '/flwr.proto.SimulationIo/GetRunStatus',
             flwr_dot_proto_dot_run__pb2.GetRunStatusRequest.SerializeToString,
             flwr_dot_proto_dot_run__pb2.GetRunStatusResponse.FromString,
-            options, channel_credentials,
-            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)
 
     @staticmethod
     def SendAppHeartbeat(request,
@@ -369,8 +480,18 @@ class SimulationIo(object):
             wait_for_ready=None,
             timeout=None,
             metadata=None):
-        return grpc.experimental.unary_unary(request, target, '/flwr.proto.SimulationIo/SendAppHeartbeat',
+        return grpc.experimental.unary_unary(
+            request,
+            target,
+            '/flwr.proto.SimulationIo/SendAppHeartbeat',
             flwr_dot_proto_dot_heartbeat__pb2.SendAppHeartbeatRequest.SerializeToString,
             flwr_dot_proto_dot_heartbeat__pb2.SendAppHeartbeatResponse.FromString,
-            options, channel_credentials,
-            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
+            options,
+            channel_credentials,
+            insecure,
+            call_credentials,
+            compression,
+            wait_for_ready,
+            timeout,
+            metadata,
+            _registered_method=True)

flwr/proto/simulationio_pb2_grpc.pyi

@@ -1,147 +1,241 @@
 """
 @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 flwr.proto.appio_pb2
 import flwr.proto.heartbeat_pb2
 import flwr.proto.log_pb2
 import flwr.proto.run_pb2
 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]
+    ...
 
 class SimulationIoStub:
-    def __init__(self, channel: grpc.Channel) -> None: ...
+    def __init__(self, channel: typing.Union[grpc.Channel, grpc.aio.Channel]) -> None: ...
     ListAppsToLaunch: grpc.UnaryUnaryMultiCallable[
         flwr.proto.appio_pb2.ListAppsToLaunchRequest,
-        flwr.proto.appio_pb2.ListAppsToLaunchResponse]
+        flwr.proto.appio_pb2.ListAppsToLaunchResponse,
+    ]
     """List runs to launch"""
 
     RequestToken: grpc.UnaryUnaryMultiCallable[
         flwr.proto.appio_pb2.RequestTokenRequest,
-        flwr.proto.appio_pb2.RequestTokenResponse]
+        flwr.proto.appio_pb2.RequestTokenResponse,
+    ]
     """Request token for a run"""
 
     GetRun: grpc.UnaryUnaryMultiCallable[
         flwr.proto.run_pb2.GetRunRequest,
-        flwr.proto.run_pb2.GetRunResponse]
+        flwr.proto.run_pb2.GetRunResponse,
+    ]
     """Get run details"""
 
     PullAppInputs: grpc.UnaryUnaryMultiCallable[
         flwr.proto.appio_pb2.PullAppInputsRequest,
-        flwr.proto.appio_pb2.PullAppInputsResponse]
+        flwr.proto.appio_pb2.PullAppInputsResponse,
+    ]
     """Pull Simulation inputs"""
 
     PushAppOutputs: grpc.UnaryUnaryMultiCallable[
         flwr.proto.appio_pb2.PushAppOutputsRequest,
-        flwr.proto.appio_pb2.PushAppOutputsResponse]
+        flwr.proto.appio_pb2.PushAppOutputsResponse,
+    ]
     """Push Simulation outputs"""
 
     UpdateRunStatus: grpc.UnaryUnaryMultiCallable[
         flwr.proto.run_pb2.UpdateRunStatusRequest,
-        flwr.proto.run_pb2.UpdateRunStatusResponse]
+        flwr.proto.run_pb2.UpdateRunStatusResponse,
+    ]
     """Update the status of a given run"""
 
     PushLogs: grpc.UnaryUnaryMultiCallable[
         flwr.proto.log_pb2.PushLogsRequest,
-        flwr.proto.log_pb2.PushLogsResponse]
+        flwr.proto.log_pb2.PushLogsResponse,
+    ]
     """Push ServerApp logs"""
 
     GetFederationOptions: grpc.UnaryUnaryMultiCallable[
         flwr.proto.run_pb2.GetFederationOptionsRequest,
-        flwr.proto.run_pb2.GetFederationOptionsResponse]
+        flwr.proto.run_pb2.GetFederationOptionsResponse,
+    ]
     """Get Federation Options"""
 
     GetRunStatus: grpc.UnaryUnaryMultiCallable[
         flwr.proto.run_pb2.GetRunStatusRequest,
-        flwr.proto.run_pb2.GetRunStatusResponse]
+        flwr.proto.run_pb2.GetRunStatusResponse,
+    ]
     """Get Run Status"""
 
     SendAppHeartbeat: grpc.UnaryUnaryMultiCallable[
         flwr.proto.heartbeat_pb2.SendAppHeartbeatRequest,
-        flwr.proto.heartbeat_pb2.SendAppHeartbeatResponse]
-    """Heartbeat"""
+        flwr.proto.heartbeat_pb2.SendAppHeartbeatResponse,
+    ]
+    """App heartbeat"""
+
+class SimulationIoAsyncStub:
+    ListAppsToLaunch: grpc.aio.UnaryUnaryMultiCallable[
+        flwr.proto.appio_pb2.ListAppsToLaunchRequest,
+        flwr.proto.appio_pb2.ListAppsToLaunchResponse,
+    ]
+    """List runs to launch"""
+
+    RequestToken: grpc.aio.UnaryUnaryMultiCallable[
+        flwr.proto.appio_pb2.RequestTokenRequest,
+        flwr.proto.appio_pb2.RequestTokenResponse,
+    ]
+    """Request token for a run"""
+
+    GetRun: grpc.aio.UnaryUnaryMultiCallable[
+        flwr.proto.run_pb2.GetRunRequest,
+        flwr.proto.run_pb2.GetRunResponse,
+    ]
+    """Get run details"""
+
+    PullAppInputs: grpc.aio.UnaryUnaryMultiCallable[
+        flwr.proto.appio_pb2.PullAppInputsRequest,
+        flwr.proto.appio_pb2.PullAppInputsResponse,
+    ]
+    """Pull Simulation inputs"""
+
+    PushAppOutputs: grpc.aio.UnaryUnaryMultiCallable[
+        flwr.proto.appio_pb2.PushAppOutputsRequest,
+        flwr.proto.appio_pb2.PushAppOutputsResponse,
+    ]
+    """Push Simulation outputs"""
+
+    UpdateRunStatus: grpc.aio.UnaryUnaryMultiCallable[
+        flwr.proto.run_pb2.UpdateRunStatusRequest,
+        flwr.proto.run_pb2.UpdateRunStatusResponse,
+    ]
+    """Update the status of a given run"""
+
+    PushLogs: grpc.aio.UnaryUnaryMultiCallable[
+        flwr.proto.log_pb2.PushLogsRequest,
+        flwr.proto.log_pb2.PushLogsResponse,
+    ]
+    """Push ServerApp logs"""
 
+    GetFederationOptions: grpc.aio.UnaryUnaryMultiCallable[
+        flwr.proto.run_pb2.GetFederationOptionsRequest,
+        flwr.proto.run_pb2.GetFederationOptionsResponse,
+    ]
+    """Get Federation Options"""
+
+    GetRunStatus: grpc.aio.UnaryUnaryMultiCallable[
+        flwr.proto.run_pb2.GetRunStatusRequest,
+        flwr.proto.run_pb2.GetRunStatusResponse,
+    ]
+    """Get Run Status"""
+
+    SendAppHeartbeat: grpc.aio.UnaryUnaryMultiCallable[
+        flwr.proto.heartbeat_pb2.SendAppHeartbeatRequest,
+        flwr.proto.heartbeat_pb2.SendAppHeartbeatResponse,
+    ]
+    """App heartbeat"""
 
 class SimulationIoServicer(metaclass=abc.ABCMeta):
     @abc.abstractmethod
-    def ListAppsToLaunch(self,
+    def ListAppsToLaunch(
+        self,
         request: flwr.proto.appio_pb2.ListAppsToLaunchRequest,
-        context: grpc.ServicerContext,
-    ) -> flwr.proto.appio_pb2.ListAppsToLaunchResponse:
+        context: _ServicerContext,
+    ) -> typing.Union[flwr.proto.appio_pb2.ListAppsToLaunchResponse, collections.abc.Awaitable[flwr.proto.appio_pb2.ListAppsToLaunchResponse]]:
         """List runs to launch"""
-        pass
 
     @abc.abstractmethod
-    def RequestToken(self,
+    def RequestToken(
+        self,
         request: flwr.proto.appio_pb2.RequestTokenRequest,
-        context: grpc.ServicerContext,
-    ) -> flwr.proto.appio_pb2.RequestTokenResponse:
+        context: _ServicerContext,
+    ) -> typing.Union[flwr.proto.appio_pb2.RequestTokenResponse, collections.abc.Awaitable[flwr.proto.appio_pb2.RequestTokenResponse]]:
         """Request token for a run"""
-        pass
 
     @abc.abstractmethod
-    def GetRun(self,
+    def GetRun(
+        self,
         request: flwr.proto.run_pb2.GetRunRequest,
-        context: grpc.ServicerContext,
-    ) -> flwr.proto.run_pb2.GetRunResponse:
+        context: _ServicerContext,
+    ) -> typing.Union[flwr.proto.run_pb2.GetRunResponse, collections.abc.Awaitable[flwr.proto.run_pb2.GetRunResponse]]:
         """Get run details"""
-        pass
 
     @abc.abstractmethod
-    def PullAppInputs(self,
+    def PullAppInputs(
+        self,
         request: flwr.proto.appio_pb2.PullAppInputsRequest,
-        context: grpc.ServicerContext,
-    ) -> flwr.proto.appio_pb2.PullAppInputsResponse:
+        context: _ServicerContext,
+    ) -> typing.Union[flwr.proto.appio_pb2.PullAppInputsResponse, collections.abc.Awaitable[flwr.proto.appio_pb2.PullAppInputsResponse]]:
         """Pull Simulation inputs"""
-        pass
 
     @abc.abstractmethod
-    def PushAppOutputs(self,
+    def PushAppOutputs(
+        self,
         request: flwr.proto.appio_pb2.PushAppOutputsRequest,
-        context: grpc.ServicerContext,
-    ) -> flwr.proto.appio_pb2.PushAppOutputsResponse:
+        context: _ServicerContext,
+    ) -> typing.Union[flwr.proto.appio_pb2.PushAppOutputsResponse, collections.abc.Awaitable[flwr.proto.appio_pb2.PushAppOutputsResponse]]:
         """Push Simulation outputs"""
-        pass
 
     @abc.abstractmethod
-    def UpdateRunStatus(self,
+    def UpdateRunStatus(
+        self,
         request: flwr.proto.run_pb2.UpdateRunStatusRequest,
-        context: grpc.ServicerContext,
-    ) -> flwr.proto.run_pb2.UpdateRunStatusResponse:
+        context: _ServicerContext,
+    ) -> typing.Union[flwr.proto.run_pb2.UpdateRunStatusResponse, collections.abc.Awaitable[flwr.proto.run_pb2.UpdateRunStatusResponse]]:
         """Update the status of a given run"""
-        pass
 
     @abc.abstractmethod
-    def PushLogs(self,
+    def PushLogs(
+        self,
         request: flwr.proto.log_pb2.PushLogsRequest,
-        context: grpc.ServicerContext,
-    ) -> flwr.proto.log_pb2.PushLogsResponse:
+        context: _ServicerContext,
+    ) -> typing.Union[flwr.proto.log_pb2.PushLogsResponse, collections.abc.Awaitable[flwr.proto.log_pb2.PushLogsResponse]]:
         """Push ServerApp logs"""
-        pass
 
     @abc.abstractmethod
-    def GetFederationOptions(self,
+    def GetFederationOptions(
+        self,
         request: flwr.proto.run_pb2.GetFederationOptionsRequest,
-        context: grpc.ServicerContext,
-    ) -> flwr.proto.run_pb2.GetFederationOptionsResponse:
+        context: _ServicerContext,
+    ) -> typing.Union[flwr.proto.run_pb2.GetFederationOptionsResponse, collections.abc.Awaitable[flwr.proto.run_pb2.GetFederationOptionsResponse]]:
         """Get Federation Options"""
-        pass
 
     @abc.abstractmethod
-    def GetRunStatus(self,
+    def GetRunStatus(
+        self,
         request: flwr.proto.run_pb2.GetRunStatusRequest,
-        context: grpc.ServicerContext,
-    ) -> flwr.proto.run_pb2.GetRunStatusResponse:
+        context: _ServicerContext,
+    ) -> typing.Union[flwr.proto.run_pb2.GetRunStatusResponse, collections.abc.Awaitable[flwr.proto.run_pb2.GetRunStatusResponse]]:
         """Get Run Status"""
-        pass
 
     @abc.abstractmethod
-    def SendAppHeartbeat(self,
+    def SendAppHeartbeat(
+        self,
         request: flwr.proto.heartbeat_pb2.SendAppHeartbeatRequest,
-        context: grpc.ServicerContext,
-    ) -> flwr.proto.heartbeat_pb2.SendAppHeartbeatResponse:
-        """Heartbeat"""
-        pass
-
+        context: _ServicerContext,
+    ) -> typing.Union[flwr.proto.heartbeat_pb2.SendAppHeartbeatResponse, collections.abc.Awaitable[flwr.proto.heartbeat_pb2.SendAppHeartbeatResponse]]:
+        """App heartbeat"""
 
-def add_SimulationIoServicer_to_server(servicer: SimulationIoServicer, server: grpc.Server) -> None: ...
+def add_SimulationIoServicer_to_server(servicer: SimulationIoServicer, server: typing.Union[grpc.Server, grpc.aio.Server]) -> None: ...