flwr-nightly 1.16.0.dev20250306__py3-none-any.whl → 1.16.0.dev20250308__py3-none-any.whl

flwr/server/superlink/linkstate/utils.py

@@ -15,30 +15,17 @@
 """Utility functions for State."""
 
 
-from logging import ERROR
 from os import urandom
-from typing import Optional, Union
+from typing import Optional
 from uuid import UUID, uuid4
 
-from flwr.common import (
-    ConfigsRecord,
-    Context,
-    Error,
-    Message,
-    Metadata,
-    log,
-    now,
-    serde,
-)
+from flwr.common import ConfigsRecord, Context, Error, Message, Metadata, now, serde
 from flwr.common.constant import SUPERLINK_NODE_ID, ErrorCode, Status, SubStatus
 from flwr.common.typing import RunStatus
 
 # pylint: disable=E0611
-from flwr.proto.error_pb2 import Error as ProtoError
 from flwr.proto.message_pb2 import Context as ProtoContext
-from flwr.proto.node_pb2 import Node
 from flwr.proto.recordset_pb2 import ConfigsRecord as ProtoConfigsRecord
-from flwr.proto.task_pb2 import Task, TaskIns, TaskRes
 
 # pylint: enable=E0611
 
@@ -244,80 +231,6 @@ def has_valid_sub_status(status: RunStatus) -> bool:
     return status.sub_status == ""
 
 
-def create_taskres_for_unavailable_taskins(taskins_id: Union[str, UUID]) -> TaskRes:
-    """Generate a TaskRes with a TaskIns unavailable error.
-
-    Parameters
-    ----------
-    taskins_id : Union[str, UUID]
-        The ID of the unavailable TaskIns.
-
-    Returns
-    -------
-    TaskRes
-        A TaskRes with an error code MESSAGE_UNAVAILABLE to indicate that the
-        inquired TaskIns ID cannot be found (due to non-existence or expiration).
-    """
-    current_time = now().timestamp()
-    return TaskRes(
-        task_id=str(uuid4()),
-        group_id="",  # Unknown group ID
-        run_id=0,  # Unknown run ID
-        task=Task(
-            # This function is only called by SuperLink, and thus it's the producer.
-            producer=Node(node_id=SUPERLINK_NODE_ID),
-            consumer=Node(node_id=SUPERLINK_NODE_ID),
-            created_at=current_time,
-            ttl=0,
-            ancestry=[str(taskins_id)],
-            task_type="",  # Unknown message type
-            error=ProtoError(
-                code=ErrorCode.MESSAGE_UNAVAILABLE,
-                reason=MESSAGE_UNAVAILABLE_ERROR_REASON,
-            ),
-        ),
-    )
-
-
-def create_taskres_for_unavailable_taskres(ref_taskins: TaskIns) -> TaskRes:
-    """Generate a TaskRes with a reply message unavailable error from a TaskIns.
-
-    Parameters
-    ----------
-    ref_taskins : TaskIns
-        The reference TaskIns object.
-
-    Returns
-    -------
-    TaskRes
-        The generated TaskRes with an error code REPLY_MESSAGE_UNAVAILABLE_ERROR_REASON,
-        indicating that the original TaskRes has expired.
-    """
-    current_time = now().timestamp()
-    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(
-            # This function is only called by SuperLink, and thus it's the producer.
-            producer=Node(node_id=SUPERLINK_NODE_ID),
-            consumer=Node(node_id=SUPERLINK_NODE_ID),
-            created_at=current_time,
-            ttl=ttl,
-            ancestry=[ref_taskins.task_id],
-            task_type=ref_taskins.task.task_type,
-            error=ProtoError(
-                code=ErrorCode.REPLY_MESSAGE_UNAVAILABLE,
-                reason=REPLY_MESSAGE_UNAVAILABLE_ERROR_REASON,
-            ),
-        ),
-    )
-
-
 def create_message_error_unavailable_res_message(ins_metadata: Metadata) -> Message:
     """Generate an error Message that the SuperLink returns carrying the specified
     error."""
@@ -366,55 +279,11 @@ def create_message_error_unavailable_ins_message(reply_to_message: UUID) -> Mess
     )
 
 
-def has_expired(task_ins_or_res: Union[TaskIns, TaskRes], current_time: float) -> bool:
-    """Check if the TaskIns/TaskRes has expired."""
-    return task_ins_or_res.task.ttl + task_ins_or_res.task.created_at < current_time
-
-
 def message_ttl_has_expired(message_metadata: Metadata, current_time: float) -> bool:
     """Check if the Message has expired."""
     return message_metadata.ttl + message_metadata.created_at < current_time
 
 
-def verify_taskins_ids(
-    inquired_taskins_ids: set[UUID],
-    found_taskins_dict: dict[UUID, TaskIns],
-    current_time: Optional[float] = None,
-    update_set: bool = True,
-) -> dict[UUID, TaskRes]:
-    """Verify found TaskIns and generate error TaskRes for invalid ones.
-
-    Parameters
-    ----------
-    inquired_taskins_ids : set[UUID]
-        Set of TaskIns IDs for which to generate error TaskRes if invalid.
-    found_taskins_dict : dict[UUID, TaskIns]
-        Dictionary containing all found TaskIns indexed by their IDs.
-    current_time : Optional[float] (default: None)
-        The current time to check for expiration. If set to `None`, the current time
-        will automatically be set to the current timestamp using `now().timestamp()`.
-    update_set : bool (default: True)
-        If True, the `inquired_taskins_ids` will be updated to remove invalid ones,
-        by default True.
-
-    Returns
-    -------
-    dict[UUID, TaskRes]
-        A dictionary of error TaskRes indexed by the corresponding TaskIns ID.
-    """
-    ret_dict = {}
-    current = current_time if current_time else now().timestamp()
-    for taskins_id in list(inquired_taskins_ids):
-        # Generate error TaskRes if the task_ins doesn't exist or has expired
-        taskins = found_taskins_dict.get(taskins_id)
-        if taskins is None or has_expired(taskins, current):
-            if update_set:
-                inquired_taskins_ids.remove(taskins_id)
-            taskres = create_taskres_for_unavailable_taskins(taskins_id)
-            ret_dict[taskins_id] = taskres
-    return ret_dict
-
-
 def verify_message_ids(
     inquired_message_ids: set[UUID],
     found_message_ins_dict: dict[UUID, Message],
@@ -457,52 +326,6 @@ def verify_message_ids(
     return ret_dict
 
 
-def verify_found_taskres(
-    inquired_taskins_ids: set[UUID],
-    found_taskins_dict: dict[UUID, TaskIns],
-    found_taskres_list: list[TaskRes],
-    current_time: Optional[float] = None,
-    update_set: bool = True,
-) -> dict[UUID, TaskRes]:
-    """Verify found TaskRes and generate error TaskRes for invalid ones.
-
-    Parameters
-    ----------
-    inquired_taskins_ids : set[UUID]
-        Set of TaskIns IDs for which to generate error TaskRes if invalid.
-    found_taskins_dict : dict[UUID, TaskIns]
-        Dictionary containing all found TaskIns indexed by their IDs.
-    found_taskres_list : dict[TaskIns, TaskRes]
-        List of found TaskRes to be verified.
-    current_time : Optional[float] (default: None)
-        The current time to check for expiration. If set to `None`, the current time
-        will automatically be set to the current timestamp using `now().timestamp()`.
-    update_set : bool (default: True)
-        If True, the `inquired_taskins_ids` will be updated to remove ones
-        that have a TaskRes, by default True.
-
-    Returns
-    -------
-    dict[UUID, TaskRes]
-        A dictionary of TaskRes indexed by the corresponding TaskIns ID.
-    """
-    ret_dict: dict[UUID, TaskRes] = {}
-    current = current_time if current_time else now().timestamp()
-    for taskres in found_taskres_list:
-        taskins_id = UUID(taskres.task.ancestry[0])
-        if update_set:
-            inquired_taskins_ids.remove(taskins_id)
-        # Check if the TaskRes has expired
-        if has_expired(taskres, current):
-            # No need to insert the error TaskRes
-            taskres = create_taskres_for_unavailable_taskres(
-                found_taskins_dict[taskins_id]
-            )
-            taskres.task.delivered_at = now().isoformat()
-        ret_dict[taskins_id] = taskres
-    return ret_dict
-
-
 def verify_found_message_replies(
     inquired_message_ids: set[UUID],
     found_message_ins_dict: dict[UUID, Message],

flwr/server/utils/__init__.py

@@ -17,10 +17,8 @@
 
 from .tensorboard import tensorboard as tensorboard
 from .validator import validate_message as validate_message
-from .validator import validate_task_ins_or_res as validate_task_ins_or_res
 
 __all__ = [
     "tensorboard",
     "validate_message",
-    "validate_task_ins_or_res",
 ]

flwr/server/utils/validator.py

@@ -16,97 +16,9 @@
 
 
 import time
-from typing import Union
 
 from flwr.common import Message
 from flwr.common.constant import SUPERLINK_NODE_ID
-from flwr.proto.task_pb2 import TaskIns, TaskRes  # pylint: disable=E0611
-
-
-# pylint: disable-next=too-many-branches,too-many-statements
-def validate_task_ins_or_res(tasks_ins_res: Union[TaskIns, TaskRes]) -> list[str]:
-    """Validate a TaskIns or TaskRes."""
-    validation_errors = []
-
-    if tasks_ins_res.task_id != "":
-        validation_errors.append("non-empty `task_id`")
-
-    if not tasks_ins_res.HasField("task"):
-        validation_errors.append("`task` does not set field `task`")
-
-    # Created/delivered/TTL/Pushed
-    if (
-        tasks_ins_res.task.created_at < 1711497600.0
-    ):  # unix timestamp of 27 March 2024 00h:00m:00s UTC
-        validation_errors.append(
-            "`created_at` must be a float that records the unix timestamp "
-            "in seconds when the message was created."
-        )
-    if tasks_ins_res.task.delivered_at != "":
-        validation_errors.append("`delivered_at` must be an empty str")
-    if tasks_ins_res.task.ttl <= 0:
-        validation_errors.append("`ttl` must be higher than zero")
-
-    # Verify TTL and created_at time
-    current_time = time.time()
-    if tasks_ins_res.task.created_at + tasks_ins_res.task.ttl <= current_time:
-        validation_errors.append("Task TTL has expired")
-
-    # TaskIns specific
-    if isinstance(tasks_ins_res, TaskIns):
-        # Task producer
-        if not tasks_ins_res.task.HasField("producer"):
-            validation_errors.append("`producer` does not set field `producer`")
-        if tasks_ins_res.task.producer.node_id != SUPERLINK_NODE_ID:
-            validation_errors.append(f"`producer.node_id` is not {SUPERLINK_NODE_ID}")
-
-        # Task consumer
-        if not tasks_ins_res.task.HasField("consumer"):
-            validation_errors.append("`consumer` does not set field `consumer`")
-        if tasks_ins_res.task.consumer.node_id == SUPERLINK_NODE_ID:
-            validation_errors.append("consumer MUST provide a valid `node_id`")
-
-        # Content check
-        if tasks_ins_res.task.task_type == "":
-            validation_errors.append("`task_type` MUST be set")
-        if not (
-            tasks_ins_res.task.HasField("recordset")
-            ^ tasks_ins_res.task.HasField("error")
-        ):
-            validation_errors.append("Either `recordset` or `error` MUST be set")
-
-        # Ancestors
-        if len(tasks_ins_res.task.ancestry) != 0:
-            validation_errors.append("`ancestry` is not empty")
-
-    # TaskRes specific
-    if isinstance(tasks_ins_res, TaskRes):
-        # Task producer
-        if not tasks_ins_res.task.HasField("producer"):
-            validation_errors.append("`producer` does not set field `producer`")
-        if tasks_ins_res.task.producer.node_id == SUPERLINK_NODE_ID:
-            validation_errors.append("producer MUST provide a valid `node_id`")
-
-        # Task consumer
-        if not tasks_ins_res.task.HasField("consumer"):
-            validation_errors.append("`consumer` does not set field `consumer`")
-        if tasks_ins_res.task.consumer.node_id != SUPERLINK_NODE_ID:
-            validation_errors.append(f"consumer is not {SUPERLINK_NODE_ID}")
-
-        # Content check
-        if tasks_ins_res.task.task_type == "":
-            validation_errors.append("`task_type` MUST be set")
-        if not (
-            tasks_ins_res.task.HasField("recordset")
-            ^ tasks_ins_res.task.HasField("error")
-        ):
-            validation_errors.append("Either `recordset` or `error` MUST be set")
-
-        # Ancestors
-        if len(tasks_ins_res.task.ancestry) == 0:
-            validation_errors.append("`ancestry` is empty")
-
-    return validation_errors
 
 
 # pylint: disable-next=too-many-branches

flwr/server/workflow/secure_aggregation/secaggplus_workflow.py

@@ -367,7 +367,7 @@ class SecAggPlusWorkflow:
 
         # Send setup configuration to clients
         cfgs_record = ConfigsRecord(sa_params_dict)  # type: ignore
-        content = RecordSet(configs_records={RECORD_KEY_CONFIGS: cfgs_record})
+        content = RecordSet({RECORD_KEY_CONFIGS: cfgs_record})
 
         def make(nid: int) -> Message:
             return driver.create_message(
@@ -417,7 +417,7 @@ class SecAggPlusWorkflow:
                 {str(nid): state.nid_to_publickeys[nid] for nid in neighbours}
             )
             cfgs_record[Key.STAGE] = Stage.SHARE_KEYS
-            content = RecordSet(configs_records={RECORD_KEY_CONFIGS: cfgs_record})
+            content = RecordSet({RECORD_KEY_CONFIGS: cfgs_record})
             return driver.create_message(
                 content=content,
                 message_type=MessageType.TRAIN,
@@ -566,7 +566,7 @@ class SecAggPlusWorkflow:
                 Key.DEAD_NODE_ID_LIST: list(neighbours & dead_nids),
             }
             cfgs_record = ConfigsRecord(cfgs_dict)  # type: ignore
-            content = RecordSet(configs_records={RECORD_KEY_CONFIGS: cfgs_record})
+            content = RecordSet({RECORD_KEY_CONFIGS: cfgs_record})
             return driver.create_message(
                 content=content,
                 message_type=MessageType.TRAIN,

flwr/superexec/exec_servicer.py

@@ -163,10 +163,10 @@ class ExecServicer(exec_pb2_grpc.ExecServicer):
         )
 
         if update_success:
-            task_ids: set[UUID] = state.get_task_ids_from_run_id(request.run_id)
+            message_ids: set[UUID] = state.get_message_ids_from_run_id(request.run_id)
 
-            # Delete TaskIns and TaskRes for the `run_id`
-            state.delete_tasks(task_ids)
+            # Delete Messages and their replies for the `run_id`
+            state.delete_messages(message_ids)
 
         return StopRunResponse(success=update_success)
 

{flwr_nightly-1.16.0.dev20250306.dist-info → flwr_nightly-1.16.0.dev20250308.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: flwr-nightly
-Version: 1.16.0.dev20250306
+Version: 1.16.0.dev20250308
 Summary: Flower: A Friendly Federated AI Framework
 Home-page: https://flower.ai
 License: Apache-2.0

{flwr_nightly-1.16.0.dev20250306.dist-info → flwr_nightly-1.16.0.dev20250308.dist-info}/RECORD

@@ -90,8 +90,7 @@ flwr/client/grpc_rere_client/connection.py,sha256=CHC5ssg7SMIN78JsWFKHg2P_fT2G_N
 flwr/client/grpc_rere_client/grpc_adapter.py,sha256=4EixBCeceM-nLu-eqB1jr0pXaGdkvoSbKmrY1ushQok,5601
 flwr/client/heartbeat.py,sha256=cx37mJBH8LyoIN4Lks85wtqT1mnU5GulQnr4pGCvAq0,2404
 flwr/client/message_handler/__init__.py,sha256=QxxQuBNpFPTHx3KiUNvQSlqMKlEnbRR1kFfc1KVje08,719
-flwr/client/message_handler/message_handler.py,sha256=s7FEfYJp5QB259Pj1L94_9AC24Kh5JyKC2U-E6eNkkY,6492
-flwr/client/message_handler/task_handler.py,sha256=zr5H2jJNKBTSI4hNGpMxbBAKUP0wSdHCJPAYjQYCxDw,1310
+flwr/client/message_handler/message_handler.py,sha256=FspUe8JmH8pROpANZVjNtxHX-07yi9A6poLZ9Ro7WK4,6492
 flwr/client/mod/__init__.py,sha256=37XeXZLFq_tzFVKVtC9JaigM2bSAU7BrGQvMPCE3Q28,1159
 flwr/client/mod/centraldp_mods.py,sha256=UGwNuqpmOWfLdfJITFgdi1TG-nLjuSb-cbEyoyfDgxQ,5415
 flwr/client/mod/comms_mods.py,sha256=s_H57ZXUPaWFifodCbRCOZW06IZRsilcOr9F-yWBtxg,2625
@@ -106,7 +105,7 @@ flwr/client/nodestate/nodestate.py,sha256=CmHZdR6kVO8tkffg42W0Yb9JdRmrUonZ9deXfU
 flwr/client/nodestate/nodestate_factory.py,sha256=apUbcJG0a_FUVsc0TkNN3q9yovc9u_J34u9iuLFKTLQ,1430
 flwr/client/numpy_client.py,sha256=chTkL9dOtK_wgUoYtzp5mfDOC1k8xPAd1qPIsB3hcjA,9581
 flwr/client/rest_client/__init__.py,sha256=5KGlp7pjc1dhNRkKlaNtUfQmg8wrRFh9lS3P3uRS-7Q,735
-flwr/client/rest_client/connection.py,sha256=9_JDGHkmaiPgk6CmcQId56Ia1ucQfrqcpNOUObdZaZU,13008
+flwr/client/rest_client/connection.py,sha256=c0mHsNXHAsbX4zNelbGd2qrV3vj90FfNE6FsiD6JzuY,12904
 flwr/client/run_info_store.py,sha256=ZN2Phi4DSLbSyzg8RmzJcVYh1g6eurHOmWRCT7GMtw4,4040
 flwr/client/supernode/__init__.py,sha256=SUhWOzcgXRNXk1V9UgB5-FaWukqqrOEajVUHEcPkwyQ,865
 flwr/client/supernode/app.py,sha256=oBbggh56HoieksBkTdDHAJC8VUFbrmGOP0-D9wdcZek,9265
@@ -131,7 +130,7 @@ flwr/common/exit/exit_code.py,sha256=PNEnCrZfOILjfDAFu5m-2YWEJBrk97xglq4zCUlqV7E
 flwr/common/exit_handlers.py,sha256=yclujry30954o0lI7vtknTajskPCvK8TXw2V3RdldXU,3174
 flwr/common/grpc.py,sha256=K60AIvIqH0CvkkiqBfw5HoxQfbFOL2DrhKPjbZ8raIE,9786
 flwr/common/logger.py,sha256=Hund1C6bEhMw3GemlzuFK22tXZ27YeHLrFB0b4LP5f8,13041
-flwr/common/message.py,sha256=VcGIBkb0te_OIKy22k4oHiOVrx0B2-8S1Gxmq0QLzTg,14210
+flwr/common/message.py,sha256=rWH_JIPnIo263-pC6Ps6XWmIgakRJA_AMBYwhE0qHcQ,14251
 flwr/common/object_ref.py,sha256=DXL8NtbN17DSYaR-Zc8WYhaG8rv0_D_cclvP7Sa66So,9134
 flwr/common/parameter.py,sha256=-bFAUayToYDF50FZGrBC1hQYJCQDtB2bbr3ZuVLMtdE,2095
 flwr/common/pyproject.py,sha256=vEAxl800XiJ1JNJDui8vuVV-08msnB6hLt7o95viZl0,1386
@@ -140,7 +139,7 @@ flwr/common/record/configsrecord.py,sha256=i40jOzBx04ysZKECwaw4FdUXMdY9HgdY8GAqK
 flwr/common/record/conversion_utils.py,sha256=ZcsM-vTm_rVtLXLFD2RY3N47V_hUr3ywTdtnpVXnOGU,1202
 flwr/common/record/metricsrecord.py,sha256=UywkEPbifiu_IyPUFoDJCi8WEVLujlqZERUWAWpc3vs,5752
 flwr/common/record/parametersrecord.py,sha256=rR0LbeNrKrdK37CiAA56Z5WBq-ZzZ2YNSUkcmr5i2lI,12950
-flwr/common/record/recordset.py,sha256=gY3nmE--8sXSIPEFMReq7nh6hMiF-sr0HpfUtiB65E8,8890
+flwr/common/record/recordset.py,sha256=ambtB74uF1pooQTfcBt1_xWsjvyDjGMc6awEt20ur2A,8547
 flwr/common/record/typeddict.py,sha256=q5hL2xkXymuiCprHWb69mUmLpWQk_XXQq0hGQ69YPaw,3599
 flwr/common/recordset_compat.py,sha256=ViSwA26h6Q55ZmV1LLjSJpcKiipV-p_JpCj4wxdE-Ow,14230
 flwr/common/retry_invoker.py,sha256=UIDKsn0AitS3fOr43WTqZAdD-TaHkBeTj1QxD7SGba0,14481
@@ -152,7 +151,7 @@ flwr/common/secure_aggregation/ndarrays_arithmetic.py,sha256=zvVAIrIyI6OSzGhpCi8
 flwr/common/secure_aggregation/quantization.py,sha256=NE_ltC3Fx5Z3bMKqJHA95wQf2tkGQlN0VZf3d1w5ABA,2400
 flwr/common/secure_aggregation/secaggplus_constants.py,sha256=9MF-oQh62uD7rt9VeNB-rHf2gBLd5GL3S9OejCxmILY,2183
 flwr/common/secure_aggregation/secaggplus_utils.py,sha256=OgYd68YBRaHQYLc-YdExj9CSpwL58bVTaPrdHoAj2AE,3214
-flwr/common/serde.py,sha256=iDVS5IXGKqEuzQAvnroH9c6KFEHxKFEWmkGP89PMOO0,31064
+flwr/common/serde.py,sha256=dgPUMPoKh8r09NERMgfnZKmEiGuKVBje7bSVFwJVZyk,27340
 flwr/common/telemetry.py,sha256=APKVubU_zJNrE-M_rip6S6Fsu41DxY3tAjFWNOgTmC0,9086
 flwr/common/typing.py,sha256=Prl8_4tKnIl_Kh5UjJGbw1tnld543EkXrX0RWffJpiA,6900
 flwr/common/version.py,sha256=aNSxLL49RKeLz8sPcZrsTEWtrAeQ0uxu6tjmfba4O60,1325
@@ -209,10 +208,6 @@ flwr/proto/simulationio_pb2.py,sha256=GOOmivAJPLQ6mPXrfKS8CyYeE6Kk-Q5mInLz5pyZSs
 flwr/proto/simulationio_pb2.pyi,sha256=oXx8_FLBe5B54wduZj-f89kub73XxNtQbThuW8YfPAs,2660
 flwr/proto/simulationio_pb2_grpc.py,sha256=9I3yAfJaeMuG-qH_5Ge45eFOftsIOmL9b8E_xHmcvKw,11232
 flwr/proto/simulationio_pb2_grpc.pyi,sha256=YHvKtyo7UdbBgdhoN0ndzZeB5vIC3JuR5PAJLrl-OKM,3206
-flwr/proto/task_pb2.py,sha256=pFKpq2QSvmFQOEpv2bgCG9jPAVauIHXQ9p5ASW-Z6lE,2322
-flwr/proto/task_pb2.pyi,sha256=PZXt-QkD-hFzTlgNVVx6SGvgbTCweBfisivEi_HfDWc,4183
-flwr/proto/task_pb2_grpc.py,sha256=1oboBPFxaTEXt9Aw7EAj8gXHDCNMhZD2VXqocC9l_gk,159
-flwr/proto/task_pb2_grpc.pyi,sha256=ff2TSiLVnG6IVQcTGzb2DIH3XRSoAvAo_RMcvbMFyc0,76
 flwr/proto/transport_pb2.py,sha256=P-jX_tUyk_8xFe-vIUUSfZlHGtk2Ou3A8eXdBKkp5AY,9824
 flwr/proto/transport_pb2.pyi,sha256=ipHQ03eFBqsxtAuAVefZ2lVr04BZ4YifJCS2eauNmy8,21627
 flwr/proto/transport_pb2_grpc.py,sha256=vLN3EHtx2aEEMCO4f1Upu-l27BPzd3-5pV-u8wPcosk,2598
@@ -225,13 +220,13 @@ flwr/server/client_proxy.py,sha256=4G-oTwhb45sfWLx2uZdcXD98IZwdTS6F88xe3akCdUg,2
 flwr/server/compat/__init__.py,sha256=VxnJtJyOjNFQXMNi9hIuzNlZM5n0Hj1p3aq_Pm2udw4,892
 flwr/server/compat/app.py,sha256=Y0xAqXLn20e6XKGb2nHdeUYdCj_Dux-SyFCc-5ivhqk,3369
 flwr/server/compat/app_utils.py,sha256=568PfvPME8KHjX-L5GB2rYQa_Wy8iUBGs22KJqn8Xk0,3824
-flwr/server/compat/driver_client_proxy.py,sha256=jMmrfqSm4-qW0jOCTpHhN9Kw0ohfMwsyKU9buBzrfG0,4958
+flwr/server/compat/driver_client_proxy.py,sha256=1ElUYvn85iMbxfyjbPiGGlmdWzNqKIIgrGJgZ18uS4E,4964
 flwr/server/compat/legacy_context.py,sha256=wBzBcfV6YO6IQGriM_FdJ5XZfiBBEEJdS_OdAiF47dY,1804
 flwr/server/criterion.py,sha256=ypbAexbztzGUxNen9RCHF91QeqiEQix4t4Ih3E-42MM,1061
 flwr/server/driver/__init__.py,sha256=bikRv6CjTwSvYh7tf10gziU5o2YotOWhhftz2tr3KDc,886
 flwr/server/driver/driver.py,sha256=X072eFWl8Kx-aZbahTkpAc1wwoojr8A4uO2yozwwSbE,5705
 flwr/server/driver/grpc_driver.py,sha256=CLa9pxN2yf1lIdxkFHVVWrs_9EFHSNUU9SJluNIQwKM,9868
-flwr/server/driver/inmemory_driver.py,sha256=ISdouBg6BvoUCMpUNykS2b6ncnCl_ZXcDGKLtpJUMLE,6614
+flwr/server/driver/inmemory_driver.py,sha256=p6p9RykDfoty94izzD4i11Xp7A8t1KUaHpbKbbVZAdU,6407
 flwr/server/history.py,sha256=qSb5_pPTrwofpSYGsZWzMPkl_4uJ4mJFWesxXDrEvDU,5026
 flwr/server/run_serverapp.py,sha256=vIPhvJx0i5sEZO4IKM6ruCXmx4ncat76rh0B4KhdhhM,2446
 flwr/server/server.py,sha256=1ZsFEptmAV-L2vP2etNC9Ed5CLSxpuKzUFkAPQ4l5Xc,17893
@@ -267,7 +262,7 @@ flwr/server/strategy/strategy.py,sha256=cXapkD5uDrt5C-RbmWDn9FLoap3Q41i7GKvbmfbC
 flwr/server/superlink/__init__.py,sha256=8tHYCfodUlRD8PCP9fHgvu8cz5N31A2QoRVL0jDJ15E,707
 flwr/server/superlink/driver/__init__.py,sha256=5soEK5QSvxNjmJQ-CGTWROc4alSAeU0e9Ad9RDhsd3E,717
 flwr/server/superlink/driver/serverappio_grpc.py,sha256=UzHwo6qYZMeOhr7nn1iZbcyDSmwvnq_kpYH0mEAndW0,2173
-flwr/server/superlink/driver/serverappio_servicer.py,sha256=hT_fM32ya3SxZFHNmfI7u2qQ_apbNHTUZB3cw6WRsmw,13116
+flwr/server/superlink/driver/serverappio_servicer.py,sha256=sZKqHOLM7sOqOlnNedwWpPNAqx5hV43bSx6eo6fbfak,12953
 flwr/server/superlink/ffs/__init__.py,sha256=FAY-zShcfPmOxosok2QyT6hTNMNctG8cH9s_nIl8jkI,840
 flwr/server/superlink/ffs/disk_ffs.py,sha256=n_Ah0sQwXGVQ9wj5965nLjdkQQbpoHCljjXKFnwftsU,3297
 flwr/server/superlink/ffs/ffs.py,sha256=qLI1UfosJugu2BKOJWqHIhafTm-YiuKqGf3OGWPH0NM,2395
@@ -281,37 +276,37 @@ flwr/server/superlink/fleet/grpc_bidi/grpc_bridge.py,sha256=JkAH_nIZaqe_9kntrg26
 flwr/server/superlink/fleet/grpc_bidi/grpc_client_proxy.py,sha256=h3EhqgelegVC4EjOXH5birmAnMoCBJcP7jpHYCnHZPk,4887
 flwr/server/superlink/fleet/grpc_bidi/grpc_server.py,sha256=mxPxyEF0IW0vV41Bqk1zfKOdRDEvXPwzJyMiRMg7nTI,5173
 flwr/server/superlink/fleet/grpc_rere/__init__.py,sha256=j2hyC342am-_Hgp1g80Y3fGDzfTI6n8QOOn2PyWf4eg,758
-flwr/server/superlink/fleet/grpc_rere/fleet_servicer.py,sha256=syWWZgsU3PHvIuvSoJxUTbEdkOtyV6psXAbRowNE_yY,5533
+flwr/server/superlink/fleet/grpc_rere/fleet_servicer.py,sha256=_J8kmdYiRoT8iUJenYic674VjX2lot8iigN3vnaWfMg,5528
 flwr/server/superlink/fleet/grpc_rere/server_interceptor.py,sha256=KPgAcnYsKPPI7_iYNSnqGbyv4tsqZuBqrTiREwVBJiM,6984
 flwr/server/superlink/fleet/message_handler/__init__.py,sha256=h8oLD7uo5lKICPy0rRdKRjTYe62u8PKkT_fA4xF5JPA,731
-flwr/server/superlink/fleet/message_handler/message_handler.py,sha256=QpLo3E-viPQ1Ia4isRM21htCfU3QH0P4wqMfXAtX8Hk,5412
+flwr/server/superlink/fleet/message_handler/message_handler.py,sha256=1pC9U-9D7Nhh9bwRZjtLxgz3osgG-tcXS-k-cAQjeF0,5242
 flwr/server/superlink/fleet/rest_rere/__init__.py,sha256=5jbYbAn75sGv-gBwOPDySE0kz96F6dTYLeMrGqNi4lM,735
 flwr/server/superlink/fleet/rest_rere/rest_api.py,sha256=L7sg9Uyp7l1bViYNWjnpaDvcG_j7FYKWC1XL-8bMpsg,6710
 flwr/server/superlink/fleet/vce/__init__.py,sha256=TZJsKTpYO_djv2EXx9Ji62I8TA0JiZF8jvRyJRZkAes,784
 flwr/server/superlink/fleet/vce/backend/__init__.py,sha256=cGgal5qpd0_MwEdIqjJbHHSIgGa6GSPoI5_byUwtG_k,1437
-flwr/server/superlink/fleet/vce/backend/backend.py,sha256=LBAQxnbfPAphVOVIvYMj0QIvVP5O-RQxKQlUGNUj974,2194
+flwr/server/superlink/fleet/vce/backend/backend.py,sha256=lr_uBMaoCp7qs59ZRpCI07wp-EsClNB-7QWFfF0r95c,2195
 flwr/server/superlink/fleet/vce/backend/raybackend.py,sha256=jsUkFEVQTnrucK1jNQ_cUM8YwL7W4MQNA1GAf8ibRdg,7156
-flwr/server/superlink/fleet/vce/vce_api.py,sha256=iXNSAJ7f8Sg1hOrVuRTGRkLdSWeFIpNWbusoOTc4I8I,12971
+flwr/server/superlink/fleet/vce/vce_api.py,sha256=xQbWzfIsXYwcEHoYxrgDcx3yGJKUoBdkc8zkJ8onzBs,12802
 flwr/server/superlink/linkstate/__init__.py,sha256=v-2JyJlCB3qyhMNwMjmcNVOq4rkooqFU0LHH8Zo1jls,1064
-flwr/server/superlink/linkstate/in_memory_linkstate.py,sha256=tw98opVU78Q9jfIORLRhwSATiMTfDa5THlcegwXSzWM,29377
-flwr/server/superlink/linkstate/linkstate.py,sha256=Yi76bAq-OJmDjl1HAQInrPIU0m9dKMiYugNJXKpKyuw,15843
+flwr/server/superlink/linkstate/in_memory_linkstate.py,sha256=Ss64nq7MgbHwx4LOER-YV7vynkzC1K4XVNk542oPg7s,21305
+flwr/server/superlink/linkstate/linkstate.py,sha256=YB3SryGNvt-bE-unYjoloJt9d3xAUPBNLK4mor8gk3M,11851
 flwr/server/superlink/linkstate/linkstate_factory.py,sha256=ISSMjDlwuN7swxjOeYlTNpI_kuZ8PGkMcJnf1dbhUSE,2069
-flwr/server/superlink/linkstate/sqlite_linkstate.py,sha256=c33n0iOZiQwNd07KfPkQbKZ7cSEX5LGg1crlUHd1ZD0,55309
-flwr/server/superlink/linkstate/utils.py,sha256=vKJYvzP_MbYD2Qo8ly6Qibyb4SI92G_4sB60ubELj64,19048
+flwr/server/superlink/linkstate/sqlite_linkstate.py,sha256=Wsx5gD6WRIMLlKarvVV1_dlS7jmfno-yTVW1-rgcIto,38276
+flwr/server/superlink/linkstate/utils.py,sha256=UunwphV90p9XzvGFJpZr3C64HgIyXjhRPNBVS3HSL8M,12696
 flwr/server/superlink/simulation/__init__.py,sha256=mg-oapC9dkzEfjXPQFior5lpWj4g9kwbLovptyYM_g0,718
 flwr/server/superlink/simulation/simulationio_grpc.py,sha256=8aUrZZLdvprKUfLLqFID4aItus9beU6m1qLQYIPB7k0,2224
 flwr/server/superlink/simulation/simulationio_servicer.py,sha256=J_TmdqM-Bxgp-iPEI3tvCuBpykw1UX0FouMQalEYAF4,6907
 flwr/server/superlink/utils.py,sha256=KVb3K_g2vYfu9TnftcN0ewmev133WZcjuEePMm8d7GE,2137
 flwr/server/typing.py,sha256=5kaRLZuxTEse9A0g7aVna2VhYxU3wTq1f3d3mtw7kXs,1019
-flwr/server/utils/__init__.py,sha256=myLB6L0ycQ0rd_nnL1pI-NLdFoOfXcj_z4gIqqj8Ye4,992
+flwr/server/utils/__init__.py,sha256=vnS9EAyVUsBOmWeYZXMoBcjeEFhqN4_KQQVk5EBiWG0,884
 flwr/server/utils/tensorboard.py,sha256=gEBD8w_5uaIfp5aw5RYH66lYZpd_SfkObHQ7eDd9MUk,5466
-flwr/server/utils/validator.py,sha256=UzYFMIyXyORXYRYXbFhOk6q6GDpOE-eIqLhlNFnR__0,7256
+flwr/server/utils/validator.py,sha256=11olMv1UyeaxLa1ci5gsJ0WALk7VwWGCQYCvLsZ56LY,3604
 flwr/server/workflow/__init__.py,sha256=SXY0XkwbkezFBxxrFB5hKUtmtAgnYISBkPouR1V71ss,902
 flwr/server/workflow/constant.py,sha256=q4DLdR8Krlxuewq2AQjwTL75hphxE5ODNz4AhViHMXk,1082
 flwr/server/workflow/default_workflows.py,sha256=RwDX7hXOI1-q4FH9A0onHvadqa5yXTzslpmShPpXodk,14152
 flwr/server/workflow/secure_aggregation/__init__.py,sha256=3XlgDOjD_hcukTGl6Bc1B-8M_dPlVSJuTbvXIbiO-Ic,880
 flwr/server/workflow/secure_aggregation/secagg_workflow.py,sha256=l2IdMdJjs1bgHs5vQgLSOVzar7v2oxUn46oCrnVE1rM,5839
-flwr/server/workflow/secure_aggregation/secaggplus_workflow.py,sha256=rfn2etO1nb7u-1oRl-H9q3enJZz3shMINZaBB7rPsC4,29671
+flwr/server/workflow/secure_aggregation/secaggplus_workflow.py,sha256=acfqyzOebe0WvXG2Z7yd7WtEtGJI3eNK2f0d_kjSdnM,29623
 flwr/simulation/__init__.py,sha256=5UcDVJNjFoSwWqHbGM1hKfTTUUNdwAtuoNvNrfvdkUY,1556
 flwr/simulation/app.py,sha256=xRVSJBnTXQUqWIYOzENfTnJlZ24CSNhWkhVEFxIu4I0,9758
 flwr/simulation/legacy_app.py,sha256=qpZI4Vvzr5TyWSLTRrMP-jN4rH2C25JI9nVSSjhFwSQ,15861
@@ -325,12 +320,12 @@ flwr/superexec/__init__.py,sha256=fcj366jh4RFby_vDwLroU4kepzqbnJgseZD_jUr_Mko,71
 flwr/superexec/app.py,sha256=Z6kYHWd62YL0Q4YKyCAbt_BcefNfbKH6V-jCC-1NkZM,1842
 flwr/superexec/deployment.py,sha256=wZ9G42gGS91knfplswh95MnQ83Fzu-rs6wcuNgDmmvY,6735
 flwr/superexec/exec_grpc.py,sha256=ttA9qoZzSLF0Mfk1L4hzOkMSNuj5rR58_kKBYwcyrAg,2864
-flwr/superexec/exec_servicer.py,sha256=72AL60LBbWD-OTxTvtPBrnb_M5rccMtU_JAYcEVQVNg,8330
+flwr/superexec/exec_servicer.py,sha256=4UpzJqPUHkBG2PZNe2lrX7XFVDOL6yw_HcoBHxuXE9A,8349
 flwr/superexec/exec_user_auth_interceptor.py,sha256=YtvcjrD2hMVmZ3y9wHuGI6FwmG_Y__q112t4fFnypy0,3793
 flwr/superexec/executor.py,sha256=_B55WW2TD1fBINpabSSDRenVHXYmvlfhv-k8hJKU4lQ,3115
 flwr/superexec/simulation.py,sha256=WQDon15oqpMopAZnwRZoTICYCfHqtkvFSqiTQ2hLD_g,4088
-flwr_nightly-1.16.0.dev20250306.dist-info/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
-flwr_nightly-1.16.0.dev20250306.dist-info/METADATA,sha256=QeUbeAdWXzaQVzTbWnix4C4edvDHIKVh_CxkJN7dTo0,15877
-flwr_nightly-1.16.0.dev20250306.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
-flwr_nightly-1.16.0.dev20250306.dist-info/entry_points.txt,sha256=JlNxX3qhaV18_2yj5a3kJW1ESxm31cal9iS_N_pf1Rk,538
-flwr_nightly-1.16.0.dev20250306.dist-info/RECORD,,
+flwr_nightly-1.16.0.dev20250308.dist-info/LICENSE,sha256=z8d0m5b2O9McPEK1xHG_dWgUBT6EfBDz6wA0F7xSPTA,11358
+flwr_nightly-1.16.0.dev20250308.dist-info/METADATA,sha256=sVPMDNkXRJ3MgQLL52bkHUFA0gTGN3e0IPeMNaQ9tvg,15877
+flwr_nightly-1.16.0.dev20250308.dist-info/WHEEL,sha256=FMvqSimYX_P7y0a7UY-_Mc83r5zkBZsCYPm7Lr0Bsq4,88
+flwr_nightly-1.16.0.dev20250308.dist-info/entry_points.txt,sha256=JlNxX3qhaV18_2yj5a3kJW1ESxm31cal9iS_N_pf1Rk,538
+flwr_nightly-1.16.0.dev20250308.dist-info/RECORD,,

flwr/client/message_handler/task_handler.py

@@ -1,37 +0,0 @@
-# Copyright 2023 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.
-# ==============================================================================
-"""Task handling."""
-
-
-from flwr.proto.task_pb2 import TaskIns  # pylint: disable=E0611
-
-
-def validate_task_ins(task_ins: TaskIns) -> bool:
-    """Validate a TaskIns before it entering the message handling process.
-
-    Parameters
-    ----------
-    task_ins: TaskIns
-        The task instruction coming from the server.
-
-    Returns
-    -------
-    is_valid: bool
-        True if the TaskIns is deemed valid and therefore suitable for
-        undergoing the message handling process, False otherwise.
-    """
-    if not (task_ins.HasField("task") and task_ins.task.HasField("recordset")):
-        return False
-    return True

flwr/proto/task_pb2.py

@@ -1,33 +0,0 @@
-# -*- coding: utf-8 -*-
-# Generated by the protocol buffer compiler.  DO NOT EDIT!
-# source: flwr/proto/task.proto
-# Protobuf Python Version: 4.25.1
-"""Generated protocol buffer code."""
-from google.protobuf import descriptor as _descriptor
-from google.protobuf import descriptor_pool as _descriptor_pool
-from google.protobuf import symbol_database as _symbol_database
-from google.protobuf.internal import builder as _builder
-# @@protoc_insertion_point(imports)
-
-_sym_db = _symbol_database.Default()
-
-
-from flwr.proto import node_pb2 as flwr_dot_proto_dot_node__pb2
-from flwr.proto import recordset_pb2 as flwr_dot_proto_dot_recordset__pb2
-from flwr.proto import error_pb2 as flwr_dot_proto_dot_error__pb2
-
-
-DESCRIPTOR = _descriptor_pool.Default().AddSerializedFile(b'\n\x15\x66lwr/proto/task.proto\x12\nflwr.proto\x1a\x15\x66lwr/proto/node.proto\x1a\x1a\x66lwr/proto/recordset.proto\x1a\x16\x66lwr/proto/error.proto\"\xf6\x01\n\x04Task\x12\"\n\x08producer\x18\x01 \x01(\x0b\x32\x10.flwr.proto.Node\x12\"\n\x08\x63onsumer\x18\x02 \x01(\x0b\x32\x10.flwr.proto.Node\x12\x12\n\ncreated_at\x18\x03 \x01(\x01\x12\x14\n\x0c\x64\x65livered_at\x18\x04 \x01(\t\x12\x0b\n\x03ttl\x18\x06 \x01(\x01\x12\x10\n\x08\x61ncestry\x18\x07 \x03(\t\x12\x11\n\ttask_type\x18\x08 \x01(\t\x12(\n\trecordset\x18\t \x01(\x0b\x32\x15.flwr.proto.RecordSet\x12 \n\x05\x65rror\x18\n \x01(\x0b\x32\x11.flwr.proto.Error\"\\\n\x07TaskIns\x12\x0f\n\x07task_id\x18\x01 \x01(\t\x12\x10\n\x08group_id\x18\x02 \x01(\t\x12\x0e\n\x06run_id\x18\x03 \x01(\x04\x12\x1e\n\x04task\x18\x04 \x01(\x0b\x32\x10.flwr.proto.Task\"\\\n\x07TaskRes\x12\x0f\n\x07task_id\x18\x01 \x01(\t\x12\x10\n\x08group_id\x18\x02 \x01(\t\x12\x0e\n\x06run_id\x18\x03 \x01(\x04\x12\x1e\n\x04task\x18\x04 \x01(\x0b\x32\x10.flwr.proto.Taskb\x06proto3')
-
-_globals = globals()
-_builder.BuildMessageAndEnumDescriptors(DESCRIPTOR, _globals)
-_builder.BuildTopDescriptorsAndMessages(DESCRIPTOR, 'flwr.proto.task_pb2', _globals)
-if _descriptor._USE_C_DESCRIPTORS == False:
-  DESCRIPTOR._options = None
-  _globals['_TASK']._serialized_start=113
-  _globals['_TASK']._serialized_end=359
-  _globals['_TASKINS']._serialized_start=361
-  _globals['_TASKINS']._serialized_end=453
-  _globals['_TASKRES']._serialized_start=455
-  _globals['_TASKRES']._serialized_end=547
-# @@protoc_insertion_point(module_scope)