flwr-nightly 1.20.0.dev20250721__tar.gz → 1.20.0.dev20250722__tar.gz

{flwr_nightly-1.20.0.dev20250721 → flwr_nightly-1.20.0.dev20250722}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.3
 Name: flwr-nightly
-Version: 1.20.0.dev20250721
+Version: 1.20.0.dev20250722
 Summary: Flower: A Friendly Federated AI Framework
 License: Apache-2.0
 Keywords: Artificial Intelligence,Federated AI,Federated Analytics,Federated Evaluation,Federated Learning,Flower,Machine Learning

{flwr_nightly-1.20.0.dev20250721 → flwr_nightly-1.20.0.dev20250722}/py/flwr/common/constant.py

@@ -123,7 +123,7 @@ AUTHZ_TYPE_YAML_KEY = "authz_type"  # For key name in YAML file
 PUBLIC_KEY_HEADER = "flwr-public-key-bin"  # Must end with "-bin" for binary data
 SIGNATURE_HEADER = "flwr-signature-bin"  # Must end with "-bin" for binary data
 TIMESTAMP_HEADER = "flwr-timestamp"
-TIMESTAMP_TOLERANCE = 10  # General tolerance for timestamp verification
+TIMESTAMP_TOLERANCE = 300  # General tolerance for timestamp verification
 SYSTEM_TIME_TOLERANCE = 5  # Allowance for system time drift
 
 # Constants for grpc retry

{flwr_nightly-1.20.0.dev20250721 → flwr_nightly-1.20.0.dev20250722}/py/flwr/common/record/array.py

@@ -264,14 +264,22 @@ class Array(InflatableObject):
 
     def slice_array(self) -> list[tuple[str, InflatableObject]]:
         """Slice Array data and construct a list of ArrayChunks."""
-        children: list[tuple[str, InflatableObject]] = []
+        # Return cached chunks if they exist
+        if "_chunks" in self.__dict__:
+            return cast(list[tuple[str, InflatableObject]], self.__dict__["_chunks"])
+
+        # Chunks are not children as some of them may be identical
+        chunks: list[tuple[str, InflatableObject]] = []
         # memoryview allows for zero-copy slicing
         data_view = memoryview(self.data)
         for start in range(0, len(data_view), MAX_ARRAY_CHUNK_SIZE):
             end = min(start + MAX_ARRAY_CHUNK_SIZE, len(data_view))
             ac = ArrayChunk(data_view[start:end])
-            children.append((ac.object_id, ac))
-        return children
+            chunks.append((ac.object_id, ac))
+
+        # Cache the chunks for future use
+        self.__dict__["_chunks"] = chunks
+        return chunks
 
     def deflate(self) -> bytes:
         """Deflate the Array."""
@@ -386,4 +394,9 @@ class Array(InflatableObject):
         if name in ("dtype", "shape", "stype", "data"):
             # Mark as dirty if any of the main attributes are set
             self.is_dirty = True
+            # Clear cached object ID
+            self.__dict__.pop("_object_id", None)
+            # Clear cached chunks if data is set
+            if name == "data":
+                self.__dict__.pop("_chunks", None)
         super().__setattr__(name, value)

{flwr_nightly-1.20.0.dev20250721 → flwr_nightly-1.20.0.dev20250722}/py/flwr/common/record/arraychunk.py

@@ -28,9 +28,6 @@ class ArrayChunk(InflatableObject):
 
     data: memoryview
 
-    def __init__(self, data: bytes) -> None:
-        self.data = memoryview(data)
-
     def deflate(self) -> bytes:
         """Deflate the ArrayChunk."""
         return add_header_to_object_body(object_body=self.data, obj=self)

{flwr_nightly-1.20.0.dev20250721 → flwr_nightly-1.20.0.dev20250722}/py/flwr/supercore/object_store/object_store.py

@@ -99,8 +99,10 @@ class ObjectStore(abc.ABC):
 
         Returns
         -------
-        bytes
-            The object stored under the given object_id.
+        Optional[bytes]
+            The object stored under the given object_id if it exists, else None.
+            The returned bytes will be b"" if the object is not yet available,
+            but has been preregistered.
         """
 
     @abc.abstractmethod

{flwr_nightly-1.20.0.dev20250721 → flwr_nightly-1.20.0.dev20250722}/py/flwr/supernode/start_client_internal.py

@@ -23,7 +23,7 @@ from contextlib import contextmanager
 from functools import partial
 from logging import INFO, WARN
 from pathlib import Path
-from typing import Callable, Optional, Union
+from typing import Callable, Optional, Union, cast
 
 import grpc
 from cryptography.hazmat.primitives.asymmetric import ec
@@ -384,23 +384,29 @@ def _push_messages(
 
         # Define the iterator for yielding object contents
         # This will yield (object_id, content) pairs
-        def yield_object_contents(_obj_tree: ObjectTree) -> Iterator[tuple[str, bytes]]:
+        def yield_object_contents(
+            _obj_tree: ObjectTree, obj_id_set: set[str]
+        ) -> Iterator[tuple[str, bytes]]:
             for tree in iterate_object_tree(_obj_tree):
-                while (content := object_store.get(tree.object_id)) is None:
+                if tree.object_id not in obj_id_set:
+                    continue
+                while (content := object_store.get(tree.object_id)) == b"":
                     # Wait for the content to be available
                     time.sleep(0.5)
-
-                yield tree.object_id, content
+                # At this point, content is guaranteed to be available
+                # therefore we can yield it after casting it to bytes
+                yield tree.object_id, cast(bytes, content)
 
         # Send the message
         try:
             # Send the reply message with its ObjectTree
-            send(message, object_tree)
+            # Get the IDs of objects to send
+            ids_obj_to_send = send(message, object_tree)
 
             # Push object contents from the ObjectStore
             run_id = message.metadata.run_id
             push_object_contents_from_iterable(
-                yield_object_contents(object_tree),
+                yield_object_contents(object_tree, ids_obj_to_send),
                 # Use functools.partial to bind run_id explicitly,
                 # avoiding late binding issues and satisfying flake8 (B023)
                 # Equivalent to:

{flwr_nightly-1.20.0.dev20250721 → flwr_nightly-1.20.0.dev20250722}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "poetry.core.masonry.api"
 
 [tool.poetry]
 name = "flwr-nightly"
-version = "1.20.0.dev20250721"
+version = "1.20.0.dev20250722"
 description = "Flower: A Friendly Federated AI Framework"
 license = "Apache-2.0"
 authors = ["The Flower Authors <hello@flower.ai>"]