syft-flwr 0.1.1__tar.gz → 0.1.3__tar.gz

{syft_flwr-0.1.1 → syft_flwr-0.1.3}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: syft-flwr
-Version: 0.1.1
+Version: 0.1.3
 Summary: syft_flwr is an open source framework that facilitate federated learning projects using Flower over the SyftBox protocol
 License-File: LICENSE
 Requires-Python: >=3.9.2
@@ -9,9 +9,9 @@ Requires-Dist: flwr[simulation]==1.17.0
 Requires-Dist: loguru>=0.7.3
 Requires-Dist: safetensors>=0.5.0
 Requires-Dist: syft-core==0.2.3
-Requires-Dist: syft-event==0.2.0
-Requires-Dist: syft-rds==0.1.0
-Requires-Dist: syft-rpc==0.2.0
+Requires-Dist: syft-event==0.2.3
+Requires-Dist: syft-rds==0.1.3
+Requires-Dist: syft-rpc==0.2.2
 Requires-Dist: tomli-w>=1.2.0
 Requires-Dist: tomli>=2.2.1
 Requires-Dist: typing-extensions>=4.13.0

{syft_flwr-0.1.1 → syft_flwr-0.1.3}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "syft-flwr"
-version = "0.1.1"
+version = "0.1.3"
 description = "syft_flwr is an open source framework that facilitate federated learning projects using Flower over the SyftBox protocol"
 readme = "README.md"
 requires-python = ">=3.9.2"
@@ -13,9 +13,9 @@ dependencies = [
     "tomli>=2.2.1",
     "tomli-w>=1.2.0",
     "syft-core==0.2.3",
-    "syft-event==0.2.0",
-    "syft-rpc==0.2.0",
-    "syft-rds==0.1.0",
+    "syft-event==0.2.3",
+    "syft-rpc==0.2.2",
+    "syft-rds==0.1.3",
 ]
 
 [project.scripts]
@@ -24,7 +24,7 @@ syft_flwr = "syft_flwr.cli:main"
 [tool.uv]
 dev-dependencies = [
     "ipykernel>=6.29.5",
-    "ipywidgets>=8.1.6",
+    "ipywidgets>=8.1.7",
     "pytest>=8.3.4",
     "pre-commit>=4.0.1",
 ]
@@ -35,7 +35,7 @@ build-backend = "hatchling.build"
 
 [tool.hatch.build.targets.wheel]
 packages = ["src/syft_flwr"]
-only-include = ["src", "pyproject.toml", "/README.md"]
+only-include = ["src"]
 exclude = ["src/**/__pycache__"]
 
 [tool.hatch.build.targets.sdist]

{syft_flwr-0.1.1 → syft_flwr-0.1.3}/src/syft_flwr/__init__.py

@@ -1,4 +1,4 @@
-__version__ = "0.1.1"
+__version__ = "0.1.3"
 
 from syft_flwr.bootstrap import bootstrap
 from syft_flwr.run import run

{syft_flwr-0.1.1 → syft_flwr-0.1.3}/src/syft_flwr/grid.py

@@ -1,16 +1,17 @@
+import os
 import time
 from typing import Iterable, cast
 
-from loguru import logger
-from syft_core import Client
-from syft_rpc import rpc, rpc_db
-from typing_extensions import Optional
-
 from flwr.common import ConfigRecord
 from flwr.common.constant import MessageType
 from flwr.common.message import Message
 from flwr.common.typing import Run
 from flwr.proto.node_pb2 import Node  # pylint: disable=E0611
+from loguru import logger
+from syft_core import Client
+from syft_rpc import rpc, rpc_db
+from typing_extensions import Optional
+
 from syft_flwr.flwr_compatibility import (
     Grid,
     RecordDict,
@@ -24,6 +25,10 @@ from syft_flwr.utils import str_to_int
 AGGREGATOR_NODE_ID = 1
 
 
+# env vars
+SYFT_FLWR_MSG_TIMEOUT = "SYFT_FLWR_MSG_TIMEOUT"
+
+
 class SyftGrid(Grid):
     def __init__(
         self,
@@ -155,7 +160,7 @@ class SyftGrid(Grid):
         self,
         messages: Iterable[Message],
         *,
-        timeout: Optional[float] = 60,
+        timeout: Optional[float] = None,
     ) -> Iterable[Message]:
         """Push messages to specified node IDs and pull the reply messages.
 
@@ -163,6 +168,17 @@ class SyftGrid(Grid):
         waits for the replies. It continues to pull replies until either all replies are
         received or the specified timeout duration (in seconds) is exceeded.
         """
+        if os.environ.get(SYFT_FLWR_MSG_TIMEOUT) is not None:
+            timeout = float(os.environ.get(SYFT_FLWR_MSG_TIMEOUT))
+        if timeout is not None:
+            logger.debug(
+                f"syft_flwr messages timeout = {timeout}: Will move on after {timeout} (s) if no reply is received"
+            )
+        else:
+            logger.debug(
+                "syft_flwr messages timeout = None: Will wait indefinitely for replies"
+            )
+
         # Push messages
         msg_ids = set(self.push_messages(messages))