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

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

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: syft-flwr
-Version: 0.1.3
+Version: 0.1.6
 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
@@ -8,10 +8,7 @@ Requires-Dist: flwr-datasets[vision]>=0.5.0
 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.3
-Requires-Dist: syft-rds==0.1.3
-Requires-Dist: syft-rpc==0.2.2
+Requires-Dist: syft-rds==0.1.5
 Requires-Dist: tomli-w>=1.2.0
 Requires-Dist: tomli>=2.2.1
 Requires-Dist: typing-extensions>=4.13.0

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

@@ -1,6 +1,6 @@
 [project]
 name = "syft-flwr"
-version = "0.1.3"
+version = "0.1.6"
 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"
@@ -12,10 +12,7 @@ dependencies = [
     "typing-extensions>=4.13.0",
     "tomli>=2.2.1",
     "tomli-w>=1.2.0",
-    "syft-core==0.2.3",
-    "syft-event==0.2.3",
-    "syft-rpc==0.2.2",
-    "syft-rds==0.1.3",
+    "syft-rds==0.1.5",
 ]
 
 [project.scripts]

syft_flwr-0.1.6/src/syft_flwr/__init__.py

@@ -0,0 +1,15 @@
+__version__ = "0.1.6"
+
+from syft_flwr.bootstrap import bootstrap
+from syft_flwr.run import run
+
+__all__ = ["bootstrap", "run"]
+
+
+# Register the mount provider for syft_rds when syft_flwr is initializedAdd commentMore actions
+from syft_rds.syft_runtime.mounts import register_mount_provider
+
+from .mounts import SyftFlwrMountProvider
+
+# Register the mount provider
+register_mount_provider("syft_flwr", SyftFlwrMountProvider())

syft_flwr-0.1.6/src/syft_flwr/mounts.py

@@ -0,0 +1,62 @@
+import json
+import os
+from pathlib import Path
+from typing import List
+
+import tomli
+from loguru import logger
+from syft_core import Client
+from syft_rds.models import DockerMount, JobConfig
+from syft_rds.syft_runtime.mounts import MountProvider
+
+
+class SyftFlwrMountProvider(MountProvider):
+    def _simplify_config(self, config_path: Path, simplified_config_path: Path) -> None:
+        """
+        Simplify the config by removing the refresh_token and setting the data_dir to /app/SyftBox
+        in order to mount the config to the container.
+        """
+        with open(config_path, "r") as fp:
+            config = json.load(fp)
+            modified_config = config.copy()
+            modified_config["data_dir"] = "/app/SyftBox"
+            modified_config.pop("refresh_token", None)
+            with open(simplified_config_path, "w") as fp:
+                json.dump(modified_config, fp)
+
+    def get_mounts(self, job_config: JobConfig) -> List[DockerMount]:
+        client = Client.load()
+        client_email = client.email
+        flwr_app_data = client.app_data("flwr")
+
+        config_path = client.config_path
+        simplified_dir = client.config_path.parent / ".simplified_configs"
+        simplified_dir.mkdir(parents=True, exist_ok=True)
+        simplified_config_path = simplified_dir / f"{client_email}.config.json"
+        self._simplify_config(config_path, simplified_config_path)
+
+        # Read app name from pyproject.toml
+        with open(job_config.function_folder / "pyproject.toml", "rb") as fp:
+            toml_dict = tomli.load(fp)
+            syft_flwr_app_name = toml_dict["tool"]["syft_flwr"]["app_name"]
+
+        rpc_messages_source = Path(f"{flwr_app_data}/{syft_flwr_app_name}/rpc/messages")
+        rpc_messages_source.mkdir(parents=True, exist_ok=True)
+        os.chmod(rpc_messages_source, 0o777)
+
+        mounts = [
+            DockerMount(
+                source=simplified_config_path,
+                target="/app/config.json",
+                mode="ro",
+            ),
+            DockerMount(
+                source=rpc_messages_source,
+                target=f"/app/SyftBox/datasites/{client_email}/app_data/flwr/{syft_flwr_app_name}/rpc/messages",
+                mode="rw",
+            ),
+        ]
+
+        logger.debug(f"Mounts: {mounts}")
+
+        return mounts

syft_flwr-0.1.3/src/syft_flwr/__init__.py

@@ -1,6 +0,0 @@
-__version__ = "0.1.3"
-
-from syft_flwr.bootstrap import bootstrap
-from syft_flwr.run import run
-
-__all__ = ["bootstrap", "run"]