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

flwr/server/superlink/linkstate/utils.py

@@ -16,7 +16,6 @@
 
 
 from os import urandom
-from typing import Optional
 
 from flwr.common import ConfigRecord, Context, Error, Message, Metadata, now, serde
 from flwr.common.constant import (
@@ -33,6 +32,7 @@ from flwr.common.typing import RunStatus
 # pylint: disable=E0611
 from flwr.proto.message_pb2 import Context as ProtoContext
 from flwr.proto.recorddict_pb2 import ConfigRecord as ProtoConfigRecord
+from flwr.supercore.utils import int64_to_uint64, uint64_to_int64
 
 # pylint: enable=E0611
 VALID_RUN_STATUS_TRANSITIONS = {
@@ -50,7 +50,8 @@ VALID_RUN_SUB_STATUSES = {
 }
 MESSAGE_UNAVAILABLE_ERROR_REASON = (
     "Error: Message Unavailable - The requested message could not be found in the "
-    "database. It may have expired due to its TTL or never existed."
+    "database. It may have expired due to its TTL, been deleted because the "
+    "destination SuperNode was removed from the federation, or never existed."
 )
 REPLY_MESSAGE_UNAVAILABLE_ERROR_REASON = (
     "Error: Reply Message Unavailable - The reply message has expired."
@@ -62,7 +63,7 @@ NODE_UNAVAILABLE_ERROR_REASON = (
 
 
 def generate_rand_int_from_bytes(
-    num_bytes: int, exclude: Optional[list[int]] = None
+    num_bytes: int, exclude: list[int] | None = None
 ) -> int:
     """Generate a random unsigned integer from `num_bytes` bytes.
 
@@ -76,58 +77,6 @@ def generate_rand_int_from_bytes(
     return num
 
 
-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:
@@ -142,7 +91,7 @@ def convert_uint64_values_in_dict_to_sint64(
     """
     for key in keys:
         if key in data_dict:
-            data_dict[key] = convert_uint64_to_sint64(data_dict[key])
+            data_dict[key] = uint64_to_int64(data_dict[key])
 
 
 def convert_sint64_values_in_dict_to_uint64(
@@ -159,7 +108,7 @@ def convert_sint64_values_in_dict_to_uint64(
     """
     for key in keys:
         if key in data_dict:
-            data_dict[key] = convert_sint64_to_uint64(data_dict[key])
+            data_dict[key] = int64_to_uint64(data_dict[key])
 
 
 def context_to_bytes(context: Context) -> bytes:
@@ -308,7 +257,7 @@ def message_ttl_has_expired(message_metadata: Metadata, current_time: float) ->
 def verify_message_ids(
     inquired_message_ids: set[str],
     found_message_ins_dict: dict[str, Message],
-    current_time: Optional[float] = None,
+    current_time: float | None = None,
     update_set: bool = True,
 ) -> dict[str, Message]:
     """Verify found Messages and generate error Messages for invalid ones.
@@ -351,7 +300,7 @@ def verify_found_message_replies(
     inquired_message_ids: set[str],
     found_message_ins_dict: dict[str, Message],
     found_message_res_list: list[Message],
-    current_time: Optional[float] = None,
+    current_time: float | None = None,
     update_set: bool = True,
 ) -> dict[str, Message]:
     """Verify found Message replies and generate error Message for invalid ones.
@@ -396,7 +345,7 @@ def check_node_availability_for_in_message(
     inquired_in_message_ids: set[str],
     found_in_message_dict: dict[str, Message],
     node_id_to_online_until: dict[int, float],
-    current_time: Optional[float] = None,
+    current_time: float | None = None,
     update_set: bool = True,
 ) -> dict[str, Message]:
     """Check node availability for given Message and generate error reply Message if

flwr/server/superlink/serverappio/serverappio_grpc.py

@@ -16,7 +16,6 @@
 
 
 from logging import INFO
-from typing import Optional
 
 import grpc
 
@@ -38,7 +37,7 @@ def run_serverappio_api_grpc(
     state_factory: LinkStateFactory,
     ffs_factory: FfsFactory,
     objectstore_factory: ObjectStoreFactory,
-    certificates: Optional[tuple[bytes, bytes, bytes]],
+    certificates: tuple[bytes, bytes, bytes] | None,
 ) -> grpc.Server:
     """Run ServerAppIo API (gRPC, request-response)."""
     # Create ServerAppIo API gRPC server

flwr/server/superlink/serverappio/serverappio_servicer.py

@@ -17,7 +17,6 @@
 
 import threading
 from logging import DEBUG, ERROR, INFO
-from typing import Optional
 
 import grpc
 
@@ -91,7 +90,6 @@ from flwr.server.superlink.utils import abort_if
 from flwr.server.utils.validator import validate_message
 from flwr.supercore.ffs import Ffs, FfsFactory
 from flwr.supercore.object_store import NoObjectInStoreError, ObjectStoreFactory
-from flwr.supercore.object_store.utils import store_mapping_and_register_objects
 
 
 class ServerAppIoServicer(serverappio_pb2_grpc.ServerAppIoServicer):
@@ -141,6 +139,13 @@ class ServerAppIoServicer(serverappio_pb2_grpc.ServerAppIoServicer):
         # Attempt to create a token for the provided run ID
         token = state.create_token(request.run_id)
 
+        # Transition the run to STARTING if token creation was successful
+        if token:
+            state.update_run_status(
+                run_id=request.run_id,
+                new_status=RunStatus(Status.STARTING, "", ""),
+            )
+
         # Return the token
         return RequestTokenResponse(token=token or "")
 
@@ -192,8 +197,11 @@ class ServerAppIoServicer(serverappio_pb2_grpc.ServerAppIoServicer):
             request_name="PushMessages",
             detail="`messages_list` must not be empty",
         )
-        message_ids: list[Optional[str]] = []
-        for message_proto in request.messages_list:
+        message_ids: list[str | None] = []
+        objects_to_push: set[str] = set()
+        for message_proto, object_tree in zip(
+            request.messages_list, request.message_object_trees, strict=True
+        ):
             message = message_from_proto(message_proto=message_proto)
             validation_errors = validate_message(message, is_reply_message=False)
             _raise_if(
@@ -206,13 +214,12 @@ class ServerAppIoServicer(serverappio_pb2_grpc.ServerAppIoServicer):
                 request_name="PushMessages",
                 detail="`Message.metadata` has mismatched `run_id`",
             )
-            # Store
-            message_id: Optional[str] = state.store_message_ins(message=message)
+            # Store objects
+            objects_to_push |= set(store.preregister(request.run_id, object_tree))
+            # Store message
+            message_id: str | None = state.store_message_ins(message=message)
             message_ids.append(message_id)
 
-        # Store Message object to descendants mapping and preregister objects
-        objects_to_push = store_mapping_and_register_objects(store, request=request)
-
         return PushAppMessagesResponse(
             message_ids=[
                 str(message_id) if message_id else "" for message_id in message_ids
@@ -316,7 +323,7 @@ class ServerAppIoServicer(serverappio_pb2_grpc.ServerAppIoServicer):
 
         ffs: Ffs = self.ffs_factory.ffs()
         if result := ffs.get(request.hash_str):
-            fab = Fab(request.hash_str, result[0])
+            fab = Fab(request.hash_str, result[0], result[1])
             return GetFabResponse(fab=fab_to_proto(fab))
 
         raise ValueError(f"Found no FAB with hash: {request.hash_str}")
@@ -343,10 +350,10 @@ class ServerAppIoServicer(serverappio_pb2_grpc.ServerAppIoServicer):
             fab = None
             if run and run.fab_hash:
                 if result := ffs.get(run.fab_hash):
-                    fab = Fab(run.fab_hash, result[0])
+                    fab = Fab(run.fab_hash, result[0], result[1])
             if run and fab and serverapp_ctxt:
-                # Update run status to STARTING
-                if state.update_run_status(run_id, RunStatus(Status.STARTING, "", "")):
+                # Update run status to RUNNING
+                if state.update_run_status(run_id, RunStatus(Status.RUNNING, "", "")):
                     log(INFO, "Starting run %d", run_id)
                     return PullAppInputsResponse(
                         context=context_to_proto(serverapp_ctxt),
@@ -355,8 +362,12 @@ class ServerAppIoServicer(serverappio_pb2_grpc.ServerAppIoServicer):
                     )
 
         # Raise an exception if the Run or Fab is not found,
-        # or if the status cannot be updated to STARTING
-        raise RuntimeError(f"Failed to start run {run_id}")
+        # or if the status cannot be updated to RUNNING
+        context.abort(
+            grpc.StatusCode.FAILED_PRECONDITION,
+            f"Failed to start run {run_id}",
+        )
+        raise RuntimeError("Unreachable code")  # for mypy
 
     def PushAppOutputs(
         self, request: PushAppOutputsRequest, context: grpc.ServicerContext
@@ -441,20 +452,14 @@ class ServerAppIoServicer(serverappio_pb2_grpc.ServerAppIoServicer):
     def SendAppHeartbeat(
         self, request: SendAppHeartbeatRequest, context: grpc.ServicerContext
     ) -> SendAppHeartbeatResponse:
-        """Handle a heartbeat from the ServerApp."""
+        """Handle a heartbeat from an app process."""
         log(DEBUG, "ServerAppIoServicer.SendAppHeartbeat")
 
         # Init state
         state = self.state_factory.state()
 
         # Acknowledge the heartbeat
-        # The app heartbeat can only be acknowledged if the run is in
-        # starting or running status.
-        success = state.acknowledge_app_heartbeat(
-            run_id=request.run_id,
-            heartbeat_interval=request.heartbeat_interval,
-        )
-
+        success = state.acknowledge_app_heartbeat(request.token)
         return SendAppHeartbeatResponse(success=success)
 
     def PushObject(

flwr/server/superlink/simulation/simulationio_grpc.py

@@ -16,7 +16,6 @@
 
 
 from logging import INFO
-from typing import Optional
 
 import grpc
 
@@ -36,7 +35,7 @@ def run_simulationio_api_grpc(
     address: str,
     state_factory: LinkStateFactory,
     ffs_factory: FfsFactory,
-    certificates: Optional[tuple[bytes, bytes, bytes]],
+    certificates: tuple[bytes, bytes, bytes] | None,
 ) -> grpc.Server:
     """Run SimulationIo API (gRPC, request-response)."""
     # Create SimulationIo API gRPC server

flwr/server/superlink/simulation/simulationio_servicer.py

@@ -110,6 +110,13 @@ class SimulationIoServicer(simulationio_pb2_grpc.SimulationIoServicer):
         # Attempt to create a token for the provided run ID
         token = state.create_token(request.run_id)
 
+        # Transition the run to STARTING if token creation was successful
+        if token:
+            state.update_run_status(
+                run_id=request.run_id,
+                new_status=RunStatus(Status.STARTING, "", ""),
+            )
+
         # Return the token
         return RequestTokenResponse(token=token or "")
 
@@ -150,10 +157,10 @@ class SimulationIoServicer(simulationio_pb2_grpc.SimulationIoServicer):
             fab = None
             if run and run.fab_hash:
                 if result := ffs.get(run.fab_hash):
-                    fab = Fab(run.fab_hash, result[0])
+                    fab = Fab(run.fab_hash, result[0], result[1])
             if run and fab and serverapp_ctxt:
-                # Update run status to STARTING
-                if state.update_run_status(run_id, RunStatus(Status.STARTING, "", "")):
+                # Update run status to RUNNING
+                if state.update_run_status(run_id, RunStatus(Status.RUNNING, "", "")):
                     log(INFO, "Starting run %d", run_id)
                     return PullAppInputsResponse(
                         context=context_to_proto(serverapp_ctxt),
@@ -162,8 +169,12 @@ class SimulationIoServicer(simulationio_pb2_grpc.SimulationIoServicer):
                     )
 
         # Raise an exception if the Run or Fab is not found,
-        # or if the status cannot be updated to STARTING
-        raise RuntimeError(f"Failed to start run {run_id}")
+        # or if the status cannot be updated to RUNNING
+        context.abort(
+            grpc.StatusCode.FAILED_PRECONDITION,
+            f"Failed to start run {run_id}",
+        )
+        raise RuntimeError("Unreachable code")  # for mypy
 
     def PushAppOutputs(
         self, request: PushAppOutputsRequest, context: ServicerContext
@@ -257,20 +268,14 @@ class SimulationIoServicer(simulationio_pb2_grpc.SimulationIoServicer):
     def SendAppHeartbeat(
         self, request: SendAppHeartbeatRequest, context: grpc.ServicerContext
     ) -> SendAppHeartbeatResponse:
-        """Handle a heartbeat from the ServerApp in simulation."""
-        log(DEBUG, "SimultionIoServicer.SendAppHeartbeat")
+        """Handle a heartbeat from an app process."""
+        log(DEBUG, "SimulationIoServicer.SendAppHeartbeat")
 
         # Init state
         state = self.state_factory.state()
 
         # Acknowledge the heartbeat
-        # The app heartbeat can only be acknowledged if the run is in
-        # starting or running status.
-        success = state.acknowledge_app_heartbeat(
-            run_id=request.run_id,
-            heartbeat_interval=request.heartbeat_interval,
-        )
-
+        success = state.acknowledge_app_heartbeat(request.token)
         return SendAppHeartbeatResponse(success=success)
 
     def _verify_token(self, token: str, context: grpc.ServicerContext) -> int:

flwr/server/superlink/utils.py

@@ -15,8 +15,6 @@
 """SuperLink utilities."""
 
 
-from typing import Optional, Union
-
 import grpc
 
 from flwr.common.constant import Status, SubStatus
@@ -36,8 +34,8 @@ def check_abort(
     run_id: int,
     abort_status_list: list[str],
     state: LinkState,
-    store: Optional[ObjectStore] = None,
-) -> Union[str, None]:
+    store: ObjectStore | None = None,
+) -> str | None:
     """Check if the status of the provided `run_id` is in `abort_status_list`."""
     run_status: RunStatus = state.get_run_status({run_id})[run_id]
 
@@ -54,7 +52,7 @@ def check_abort(
     return None
 
 
-def abort_grpc_context(msg: Union[str, None], context: grpc.ServicerContext) -> None:
+def abort_grpc_context(msg: str | None, context: grpc.ServicerContext) -> None:
     """Abort context with statuscode PERMISSION_DENIED if `msg` is not None."""
     if msg is not None:
         context.abort(grpc.StatusCode.PERMISSION_DENIED, msg)
@@ -64,7 +62,7 @@ def abort_if(
     run_id: int,
     abort_status_list: list[str],
     state: LinkState,
-    store: Optional[ObjectStore],
+    store: ObjectStore | None,
     context: grpc.ServicerContext,
 ) -> None:
     """Abort context if status of the provided `run_id` is in `abort_status_list`."""

flwr/server/typing.py

@@ -15,7 +15,7 @@
 """Custom types for Flower servers."""
 
 
-from typing import Callable
+from collections.abc import Callable
 
 from flwr.common import Context
 

flwr/server/utils/tensorboard.py

@@ -16,20 +16,16 @@
 
 
 import os
+from collections.abc import Callable
 from datetime import datetime
 from logging import WARN
-from typing import Callable, Optional, Union, cast
+from typing import cast
 
 from flwr.common import EvaluateRes, Scalar
 from flwr.common.logger import log
 from flwr.server.client_proxy import ClientProxy
 from flwr.server.strategy import Strategy
 
-try:
-    import tensorflow as TF
-except ModuleNotFoundError:
-    TF = None
-
 MISSING_EXTRA_TF = """
 Extra dependency required for using tensorboard are missing.
 The program will continue without tensorboard.
@@ -59,6 +55,17 @@ def tensorboard(logdir: str) -> Callable[[Strategy], Strategy]:
         # Variant 2
         strategy = tensorboard(logdir=LOGDIR)(FedAvg)()
     """
+    log(
+        WARN,
+        "The `tensorboard` function is deprecated and will be removed "
+        "in a future release.",
+    )
+    # Lazy import of TensorFlow to avoid slow import times
+    try:
+        import tensorflow as TF  # pylint: disable=import-outside-toplevel
+    except ModuleNotFoundError:
+        TF = None  # pylint: disable=invalid-name
+
     print(
         "\n\t\033[32mStart TensorBoard with the following parameters"
         f"\n\t$ tensorboard --logdir {logdir}\033[39m\n"
@@ -93,8 +100,8 @@ def tensorboard(logdir: str) -> Callable[[Strategy], Strategy]:
                 self,
                 server_round: int,
                 results: list[tuple[ClientProxy, EvaluateRes]],
-                failures: list[Union[tuple[ClientProxy, EvaluateRes], BaseException]],
-            ) -> tuple[Optional[float], dict[str, Scalar]]:
+                failures: list[tuple[ClientProxy, EvaluateRes] | BaseException],
+            ) -> tuple[float | None, dict[str, Scalar]]:
                 """Hooks into aggregate_evaluate for TensorBoard logging purpose."""
                 # Execute decorated function and extract results for logging
                 # They will be returned at the end of this function but also

flwr/server/utils/validator.py

@@ -15,10 +15,9 @@
 """Validators."""
 
 
-import time
-
 from flwr.common import Message
 from flwr.common.constant import SUPERLINK_NODE_ID
+from flwr.common.date import now
 
 
 # pylint: disable-next=too-many-branches
@@ -44,7 +43,7 @@ def validate_message(message: Message, is_reply_message: bool) -> list[str]:
         validation_errors.append("`metadata.ttl` must be higher than zero")
 
     # Verify TTL and created_at time
-    current_time = time.time()
+    current_time = now().timestamp()
     if metadata.created_at + metadata.ttl <= current_time:
         validation_errors.append("Message TTL has expired")
 

flwr/server/workflow/default_workflows.py

@@ -18,7 +18,7 @@
 import io
 import timeit
 from logging import INFO, WARN
-from typing import Optional, Union, cast
+from typing import cast
 
 import flwr.common.recorddict_compat as compat
 from flwr.common import (
@@ -47,8 +47,8 @@ class DefaultWorkflow:
 
     def __init__(
         self,
-        fit_workflow: Optional[Workflow] = None,
-        evaluate_workflow: Optional[Workflow] = None,
+        fit_workflow: Workflow | None = None,
+        evaluate_workflow: Workflow | None = None,
     ) -> None:
         if fit_workflow is None:
             fit_workflow = default_fit_workflow
@@ -275,7 +275,7 @@ def default_fit_workflow(grid: Grid, context: Context) -> None:  # pylint: disab
 
     # Aggregate training results
     results: list[tuple[ClientProxy, FitRes]] = []
-    failures: list[Union[tuple[ClientProxy, FitRes], BaseException]] = []
+    failures: list[tuple[ClientProxy, FitRes] | BaseException] = []
     for msg in messages:
         if msg.has_content():
             proxy = node_id_to_proxy[msg.metadata.src_node_id]
@@ -357,7 +357,7 @@ def default_evaluate_workflow(grid: Grid, context: Context) -> None:
 
     # Aggregate the evaluation results
     results: list[tuple[ClientProxy, EvaluateRes]] = []
-    failures: list[Union[tuple[ClientProxy, EvaluateRes], BaseException]] = []
+    failures: list[tuple[ClientProxy, EvaluateRes] | BaseException] = []
     for msg in messages:
         if msg.has_content():
             proxy = node_id_to_proxy[msg.metadata.src_node_id]

flwr/server/workflow/secure_aggregation/secagg_workflow.py

@@ -15,8 +15,6 @@
 """Workflow for the SecAgg protocol."""
 
 
-from typing import Optional, Union
-
 from .secaggplus_workflow import SecAggPlusWorkflow
 
 
@@ -94,13 +92,13 @@ class SecAggWorkflow(SecAggPlusWorkflow):
 
     def __init__(  # pylint: disable=R0913
         self,
-        reconstruction_threshold: Union[int, float],
+        reconstruction_threshold: int | float,
         *,
         max_weight: float = 1000.0,
         clipping_range: float = 8.0,
         quantization_range: int = 4194304,
         modulus_range: int = 4294967296,
-        timeout: Optional[float] = None,
+        timeout: float | None = None,
     ) -> None:
         super().__init__(
             num_shares=1.0,

flwr/server/workflow/secure_aggregation/secaggplus_workflow.py

@@ -18,7 +18,7 @@
 import random
 from dataclasses import dataclass, field
 from logging import DEBUG, ERROR, INFO, WARN
-from typing import Optional, Union, cast
+from typing import cast
 
 import flwr.common.recorddict_compat as compat
 from flwr.common import (
@@ -35,8 +35,6 @@ from flwr.common import (
 )
 from flwr.common.secure_aggregation.crypto.shamir import combine_shares
 from flwr.common.secure_aggregation.crypto.symmetric_encryption import (
-    bytes_to_private_key,
-    bytes_to_public_key,
     generate_shared_key,
 )
 from flwr.common.secure_aggregation.ndarrays_arithmetic import (
@@ -56,6 +54,10 @@ from flwr.common.secure_aggregation.secaggplus_utils import pseudo_rand_gen
 from flwr.server.client_proxy import ClientProxy
 from flwr.server.compat.legacy_context import LegacyContext
 from flwr.server.grid import Grid
+from flwr.supercore.primitives.asymmetric import (
+    bytes_to_private_key,
+    bytes_to_public_key,
+)
 
 from ..constant import MAIN_CONFIGS_RECORD, MAIN_PARAMS_RECORD
 from ..constant import Key as WorkflowKey
@@ -167,14 +169,14 @@ class SecAggPlusWorkflow:
 
     def __init__(  # pylint: disable=R0913
         self,
-        num_shares: Union[int, float],
-        reconstruction_threshold: Union[int, float],
+        num_shares: int | float,
+        reconstruction_threshold: int | float,
         *,
         max_weight: float = 1000.0,
         clipping_range: float = 8.0,
         quantization_range: int = 4194304,
         modulus_range: int = 4294967296,
-        timeout: Optional[float] = None,
+        timeout: float | None = None,
     ) -> None:
         self.num_shares = num_shares
         self.reconstruction_threshold = reconstruction_threshold
@@ -209,7 +211,7 @@ class SecAggPlusWorkflow:
 
     def _check_init_params(self) -> None:  # pylint: disable=R0912
         # Check `num_shares`
-        if not isinstance(self.num_shares, (int, float)):
+        if not isinstance(self.num_shares, (int | float)):
             raise TypeError("`num_shares` must be of type int or float.")
         if isinstance(self.num_shares, int):
             if self.num_shares == 1:
@@ -227,7 +229,7 @@ class SecAggPlusWorkflow:
             raise ValueError("`num_shares` as a float must be greater than 0.")
 
         # Check `reconstruction_threshold`
-        if not isinstance(self.reconstruction_threshold, (int, float)):
+        if not isinstance(self.reconstruction_threshold, (int | float)):
             raise TypeError("`reconstruction_threshold` must be of type int or float.")
         if isinstance(self.reconstruction_threshold, int):
             if self.reconstruction_threshold == 1:
@@ -465,7 +467,7 @@ class SecAggPlusWorkflow:
             dsts += dst_lst
             ciphertexts += ctxt_lst
 
-        for src, dst, ciphertext in zip(srcs, dsts, ciphertexts):
+        for src, dst, ciphertext in zip(srcs, dsts, ciphertexts, strict=True):
             if dst in fwd_ciphertexts:
                 fwd_ciphertexts[dst].append(ciphertext)
                 fwd_srcs[dst].append(src)
@@ -602,7 +604,7 @@ class SecAggPlusWorkflow:
             res_dict = msg.content.config_records[RECORD_KEY_CONFIGS]
             nids = cast(list[int], res_dict[Key.NODE_ID_LIST])
             shares = cast(list[bytes], res_dict[Key.SHARE_LIST])
-            for owner_nid, share in zip(nids, shares):
+            for owner_nid, share in zip(nids, shares, strict=True):
                 collected_shares_dict[owner_nid].append(share)
 
         # Remove masks for every active client after collect_masked_vectors stage