flwr 1.12.0__py3-none-any.whl → 1.13.1__py3-none-any.whl

flwr/proto/driver_pb2_grpc.py

@@ -1,239 +0,0 @@
-# Generated by the gRPC Python protocol compiler plugin. DO NOT EDIT!
-"""Client and server classes corresponding to protobuf-defined services."""
-import grpc
-
-from flwr.proto import driver_pb2 as flwr_dot_proto_dot_driver__pb2
-from flwr.proto import fab_pb2 as flwr_dot_proto_dot_fab__pb2
-from flwr.proto import run_pb2 as flwr_dot_proto_dot_run__pb2
-
-
-class DriverStub(object):
-    """Missing associated documentation comment in .proto file."""
-
-    def __init__(self, channel):
-        """Constructor.
-
-        Args:
-            channel: A grpc.Channel.
-        """
-        self.CreateRun = channel.unary_unary(
-                '/flwr.proto.Driver/CreateRun',
-                request_serializer=flwr_dot_proto_dot_run__pb2.CreateRunRequest.SerializeToString,
-                response_deserializer=flwr_dot_proto_dot_run__pb2.CreateRunResponse.FromString,
-                )
-        self.GetNodes = channel.unary_unary(
-                '/flwr.proto.Driver/GetNodes',
-                request_serializer=flwr_dot_proto_dot_driver__pb2.GetNodesRequest.SerializeToString,
-                response_deserializer=flwr_dot_proto_dot_driver__pb2.GetNodesResponse.FromString,
-                )
-        self.PushTaskIns = channel.unary_unary(
-                '/flwr.proto.Driver/PushTaskIns',
-                request_serializer=flwr_dot_proto_dot_driver__pb2.PushTaskInsRequest.SerializeToString,
-                response_deserializer=flwr_dot_proto_dot_driver__pb2.PushTaskInsResponse.FromString,
-                )
-        self.PullTaskRes = channel.unary_unary(
-                '/flwr.proto.Driver/PullTaskRes',
-                request_serializer=flwr_dot_proto_dot_driver__pb2.PullTaskResRequest.SerializeToString,
-                response_deserializer=flwr_dot_proto_dot_driver__pb2.PullTaskResResponse.FromString,
-                )
-        self.GetRun = channel.unary_unary(
-                '/flwr.proto.Driver/GetRun',
-                request_serializer=flwr_dot_proto_dot_run__pb2.GetRunRequest.SerializeToString,
-                response_deserializer=flwr_dot_proto_dot_run__pb2.GetRunResponse.FromString,
-                )
-        self.GetFab = channel.unary_unary(
-                '/flwr.proto.Driver/GetFab',
-                request_serializer=flwr_dot_proto_dot_fab__pb2.GetFabRequest.SerializeToString,
-                response_deserializer=flwr_dot_proto_dot_fab__pb2.GetFabResponse.FromString,
-                )
-
-
-class DriverServicer(object):
-    """Missing associated documentation comment in .proto file."""
-
-    def CreateRun(self, request, context):
-        """Request run_id
-        """
-        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
-        context.set_details('Method not implemented!')
-        raise NotImplementedError('Method not implemented!')
-
-    def GetNodes(self, request, context):
-        """Return a set of nodes
-        """
-        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
-        context.set_details('Method not implemented!')
-        raise NotImplementedError('Method not implemented!')
-
-    def PushTaskIns(self, request, context):
-        """Create one or more tasks
-        """
-        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
-        context.set_details('Method not implemented!')
-        raise NotImplementedError('Method not implemented!')
-
-    def PullTaskRes(self, request, context):
-        """Get task results
-        """
-        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
-        context.set_details('Method not implemented!')
-        raise NotImplementedError('Method not implemented!')
-
-    def GetRun(self, request, context):
-        """Get run details
-        """
-        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
-        context.set_details('Method not implemented!')
-        raise NotImplementedError('Method not implemented!')
-
-    def GetFab(self, request, context):
-        """Get FAB
-        """
-        context.set_code(grpc.StatusCode.UNIMPLEMENTED)
-        context.set_details('Method not implemented!')
-        raise NotImplementedError('Method not implemented!')
-
-
-def add_DriverServicer_to_server(servicer, server):
-    rpc_method_handlers = {
-            'CreateRun': grpc.unary_unary_rpc_method_handler(
-                    servicer.CreateRun,
-                    request_deserializer=flwr_dot_proto_dot_run__pb2.CreateRunRequest.FromString,
-                    response_serializer=flwr_dot_proto_dot_run__pb2.CreateRunResponse.SerializeToString,
-            ),
-            'GetNodes': grpc.unary_unary_rpc_method_handler(
-                    servicer.GetNodes,
-                    request_deserializer=flwr_dot_proto_dot_driver__pb2.GetNodesRequest.FromString,
-                    response_serializer=flwr_dot_proto_dot_driver__pb2.GetNodesResponse.SerializeToString,
-            ),
-            'PushTaskIns': grpc.unary_unary_rpc_method_handler(
-                    servicer.PushTaskIns,
-                    request_deserializer=flwr_dot_proto_dot_driver__pb2.PushTaskInsRequest.FromString,
-                    response_serializer=flwr_dot_proto_dot_driver__pb2.PushTaskInsResponse.SerializeToString,
-            ),
-            'PullTaskRes': grpc.unary_unary_rpc_method_handler(
-                    servicer.PullTaskRes,
-                    request_deserializer=flwr_dot_proto_dot_driver__pb2.PullTaskResRequest.FromString,
-                    response_serializer=flwr_dot_proto_dot_driver__pb2.PullTaskResResponse.SerializeToString,
-            ),
-            'GetRun': grpc.unary_unary_rpc_method_handler(
-                    servicer.GetRun,
-                    request_deserializer=flwr_dot_proto_dot_run__pb2.GetRunRequest.FromString,
-                    response_serializer=flwr_dot_proto_dot_run__pb2.GetRunResponse.SerializeToString,
-            ),
-            'GetFab': grpc.unary_unary_rpc_method_handler(
-                    servicer.GetFab,
-                    request_deserializer=flwr_dot_proto_dot_fab__pb2.GetFabRequest.FromString,
-                    response_serializer=flwr_dot_proto_dot_fab__pb2.GetFabResponse.SerializeToString,
-            ),
-    }
-    generic_handler = grpc.method_handlers_generic_handler(
-            'flwr.proto.Driver', rpc_method_handlers)
-    server.add_generic_rpc_handlers((generic_handler,))
-
-
- # This class is part of an EXPERIMENTAL API.
-class Driver(object):
-    """Missing associated documentation comment in .proto file."""
-
-    @staticmethod
-    def CreateRun(request,
-            target,
-            options=(),
-            channel_credentials=None,
-            call_credentials=None,
-            insecure=False,
-            compression=None,
-            wait_for_ready=None,
-            timeout=None,
-            metadata=None):
-        return grpc.experimental.unary_unary(request, target, '/flwr.proto.Driver/CreateRun',
-            flwr_dot_proto_dot_run__pb2.CreateRunRequest.SerializeToString,
-            flwr_dot_proto_dot_run__pb2.CreateRunResponse.FromString,
-            options, channel_credentials,
-            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
-
-    @staticmethod
-    def GetNodes(request,
-            target,
-            options=(),
-            channel_credentials=None,
-            call_credentials=None,
-            insecure=False,
-            compression=None,
-            wait_for_ready=None,
-            timeout=None,
-            metadata=None):
-        return grpc.experimental.unary_unary(request, target, '/flwr.proto.Driver/GetNodes',
-            flwr_dot_proto_dot_driver__pb2.GetNodesRequest.SerializeToString,
-            flwr_dot_proto_dot_driver__pb2.GetNodesResponse.FromString,
-            options, channel_credentials,
-            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
-
-    @staticmethod
-    def PushTaskIns(request,
-            target,
-            options=(),
-            channel_credentials=None,
-            call_credentials=None,
-            insecure=False,
-            compression=None,
-            wait_for_ready=None,
-            timeout=None,
-            metadata=None):
-        return grpc.experimental.unary_unary(request, target, '/flwr.proto.Driver/PushTaskIns',
-            flwr_dot_proto_dot_driver__pb2.PushTaskInsRequest.SerializeToString,
-            flwr_dot_proto_dot_driver__pb2.PushTaskInsResponse.FromString,
-            options, channel_credentials,
-            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
-
-    @staticmethod
-    def PullTaskRes(request,
-            target,
-            options=(),
-            channel_credentials=None,
-            call_credentials=None,
-            insecure=False,
-            compression=None,
-            wait_for_ready=None,
-            timeout=None,
-            metadata=None):
-        return grpc.experimental.unary_unary(request, target, '/flwr.proto.Driver/PullTaskRes',
-            flwr_dot_proto_dot_driver__pb2.PullTaskResRequest.SerializeToString,
-            flwr_dot_proto_dot_driver__pb2.PullTaskResResponse.FromString,
-            options, channel_credentials,
-            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)
-
-    @staticmethod
-    def GetRun(request,
-            target,
-            options=(),
-            channel_credentials=None,
-            call_credentials=None,
-            insecure=False,
-            compression=None,
-            wait_for_ready=None,
-            timeout=None,
-            metadata=None):
-        return grpc.experimental.unary_unary(request, target, '/flwr.proto.Driver/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)
-
-    @staticmethod
-    def GetFab(request,
-            target,
-            options=(),
-            channel_credentials=None,
-            call_credentials=None,
-            insecure=False,
-            compression=None,
-            wait_for_ready=None,
-            timeout=None,
-            metadata=None):
-        return grpc.experimental.unary_unary(request, target, '/flwr.proto.Driver/GetFab',
-            flwr_dot_proto_dot_fab__pb2.GetFabRequest.SerializeToString,
-            flwr_dot_proto_dot_fab__pb2.GetFabResponse.FromString,
-            options, channel_credentials,
-            insecure, call_credentials, compression, wait_for_ready, timeout, metadata)

flwr/proto/driver_pb2_grpc.pyi

@@ -1,94 +0,0 @@
-"""
-@generated by mypy-protobuf.  Do not edit manually!
-isort:skip_file
-"""
-import abc
-import flwr.proto.driver_pb2
-import flwr.proto.fab_pb2
-import flwr.proto.run_pb2
-import grpc
-
-class DriverStub:
-    def __init__(self, channel: grpc.Channel) -> None: ...
-    CreateRun: grpc.UnaryUnaryMultiCallable[
-        flwr.proto.run_pb2.CreateRunRequest,
-        flwr.proto.run_pb2.CreateRunResponse]
-    """Request run_id"""
-
-    GetNodes: grpc.UnaryUnaryMultiCallable[
-        flwr.proto.driver_pb2.GetNodesRequest,
-        flwr.proto.driver_pb2.GetNodesResponse]
-    """Return a set of nodes"""
-
-    PushTaskIns: grpc.UnaryUnaryMultiCallable[
-        flwr.proto.driver_pb2.PushTaskInsRequest,
-        flwr.proto.driver_pb2.PushTaskInsResponse]
-    """Create one or more tasks"""
-
-    PullTaskRes: grpc.UnaryUnaryMultiCallable[
-        flwr.proto.driver_pb2.PullTaskResRequest,
-        flwr.proto.driver_pb2.PullTaskResResponse]
-    """Get task results"""
-
-    GetRun: grpc.UnaryUnaryMultiCallable[
-        flwr.proto.run_pb2.GetRunRequest,
-        flwr.proto.run_pb2.GetRunResponse]
-    """Get run details"""
-
-    GetFab: grpc.UnaryUnaryMultiCallable[
-        flwr.proto.fab_pb2.GetFabRequest,
-        flwr.proto.fab_pb2.GetFabResponse]
-    """Get FAB"""
-
-
-class DriverServicer(metaclass=abc.ABCMeta):
-    @abc.abstractmethod
-    def CreateRun(self,
-        request: flwr.proto.run_pb2.CreateRunRequest,
-        context: grpc.ServicerContext,
-    ) -> flwr.proto.run_pb2.CreateRunResponse:
-        """Request run_id"""
-        pass
-
-    @abc.abstractmethod
-    def GetNodes(self,
-        request: flwr.proto.driver_pb2.GetNodesRequest,
-        context: grpc.ServicerContext,
-    ) -> flwr.proto.driver_pb2.GetNodesResponse:
-        """Return a set of nodes"""
-        pass
-
-    @abc.abstractmethod
-    def PushTaskIns(self,
-        request: flwr.proto.driver_pb2.PushTaskInsRequest,
-        context: grpc.ServicerContext,
-    ) -> flwr.proto.driver_pb2.PushTaskInsResponse:
-        """Create one or more tasks"""
-        pass
-
-    @abc.abstractmethod
-    def PullTaskRes(self,
-        request: flwr.proto.driver_pb2.PullTaskResRequest,
-        context: grpc.ServicerContext,
-    ) -> flwr.proto.driver_pb2.PullTaskResResponse:
-        """Get task results"""
-        pass
-
-    @abc.abstractmethod
-    def GetRun(self,
-        request: flwr.proto.run_pb2.GetRunRequest,
-        context: grpc.ServicerContext,
-    ) -> flwr.proto.run_pb2.GetRunResponse:
-        """Get run details"""
-        pass
-
-    @abc.abstractmethod
-    def GetFab(self,
-        request: flwr.proto.fab_pb2.GetFabRequest,
-        context: grpc.ServicerContext,
-    ) -> flwr.proto.fab_pb2.GetFabResponse:
-        """Get FAB"""
-        pass
-
-
-def add_DriverServicer_to_server(servicer: DriverServicer, server: grpc.Server) -> None: ...

flwr/server/superlink/state/utils.py

@@ -1,148 +0,0 @@
-# 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.
-# ==============================================================================
-"""Utility functions for State."""
-
-
-import time
-from logging import ERROR
-from os import urandom
-from uuid import uuid4
-
-from flwr.common import log
-from flwr.common.constant import ErrorCode
-from flwr.proto.error_pb2 import Error  # pylint: disable=E0611
-from flwr.proto.node_pb2 import Node  # pylint: disable=E0611
-from flwr.proto.task_pb2 import Task, TaskIns, TaskRes  # pylint: disable=E0611
-
-NODE_UNAVAILABLE_ERROR_REASON = (
-    "Error: Node Unavailable - The destination node is currently unavailable. "
-    "It exceeds the time limit specified in its last ping."
-)
-
-
-def generate_rand_int_from_bytes(num_bytes: int) -> int:
-    """Generate a random unsigned integer from `num_bytes` bytes."""
-    return int.from_bytes(urandom(num_bytes), "little", signed=False)
-
-
-def convert_uint64_to_sint64(u: int) -> int:
-    """Convert a uint64 value to a sint64 value with the same bit sequence.
-
-    Parameters
-    ----------
-    u : int
-        The unsigned 64-bit integer to convert.
-
-    Returns
-    -------
-    int
-        The signed 64-bit integer equivalent.
-
-        The signed 64-bit integer will have the same bit pattern as the
-        unsigned 64-bit integer but may have a different decimal value.
-
-        For numbers within the range [0, `sint64` max value], the decimal
-        value remains the same. However, for numbers greater than the `sint64`
-        max value, the decimal value will differ due to the wraparound caused
-        by the sign bit.
-    """
-    if u >= (1 << 63):
-        return u - (1 << 64)
-    return u
-
-
-def convert_sint64_to_uint64(s: int) -> int:
-    """Convert a sint64 value to a uint64 value with the same bit sequence.
-
-    Parameters
-    ----------
-    s : int
-        The signed 64-bit integer to convert.
-
-    Returns
-    -------
-    int
-        The unsigned 64-bit integer equivalent.
-
-        The unsigned 64-bit integer will have the same bit pattern as the
-        signed 64-bit integer but may have a different decimal value.
-
-        For negative `sint64` values, the conversion adds 2^64 to the
-        signed value to obtain the equivalent `uint64` value. For non-negative
-        `sint64` values, the decimal value remains unchanged in the `uint64`
-        representation.
-    """
-    if s < 0:
-        return s + (1 << 64)
-    return s
-
-
-def convert_uint64_values_in_dict_to_sint64(
-    data_dict: dict[str, int], keys: list[str]
-) -> None:
-    """Convert uint64 values to sint64 in the given dictionary.
-
-    Parameters
-    ----------
-    data_dict : dict[str, int]
-        A dictionary where the values are integers to be converted.
-    keys : list[str]
-        A list of keys in the dictionary whose values need to be converted.
-    """
-    for key in keys:
-        if key in data_dict:
-            data_dict[key] = convert_uint64_to_sint64(data_dict[key])
-
-
-def convert_sint64_values_in_dict_to_uint64(
-    data_dict: dict[str, int], keys: list[str]
-) -> None:
-    """Convert sint64 values to uint64 in the given dictionary.
-
-    Parameters
-    ----------
-    data_dict : dict[str, int]
-        A dictionary where the values are integers to be converted.
-    keys : list[str]
-        A list of keys in the dictionary whose values need to be converted.
-    """
-    for key in keys:
-        if key in data_dict:
-            data_dict[key] = convert_sint64_to_uint64(data_dict[key])
-
-
-def make_node_unavailable_taskres(ref_taskins: TaskIns) -> TaskRes:
-    """Generate a TaskRes with a node unavailable error from a TaskIns."""
-    current_time = time.time()
-    ttl = ref_taskins.task.ttl - (current_time - ref_taskins.task.created_at)
-    if ttl < 0:
-        log(ERROR, "Creating TaskRes for TaskIns that exceeds its TTL.")
-        ttl = 0
-    return TaskRes(
-        task_id=str(uuid4()),
-        group_id=ref_taskins.group_id,
-        run_id=ref_taskins.run_id,
-        task=Task(
-            producer=Node(node_id=ref_taskins.task.consumer.node_id, anonymous=False),
-            consumer=Node(node_id=ref_taskins.task.producer.node_id, anonymous=False),
-            created_at=current_time,
-            ttl=ttl,
-            ancestry=[ref_taskins.task_id],
-            task_type=ref_taskins.task.task_type,
-            error=Error(
-                code=ErrorCode.NODE_UNAVAILABLE, reason=NODE_UNAVAILABLE_ERROR_REASON
-            ),
-        ),
-    )