MEDfl 2.0.5.dev3__py3-none-any.whl → 2.0.5.dev6__py3-none-any.whl

MEDfl/NetManager/database_connector.py

@@ -3,6 +3,7 @@ import subprocess , sys
 from sqlalchemy import create_engine
 from configparser import ConfigParser
 
+
 class DatabaseManager:
     def __init__(self):
         from MEDfl.LearningManager.utils import load_db_config

MEDfl/rw/client.py

@@ -35,10 +35,10 @@ class DPConfig:
 
     def __init__(
         self,
-        noise_multiplier: float = 1.0,
-        max_grad_norm: float = 1.0,
-        batch_size: int = 32,
-        secure_rng: bool = False,
+        noise_multiplier=1.0,
+        max_grad_norm=1.0,
+        batch_size=32,
+        secure_rng=False,
     ):
         self.noise_multiplier = noise_multiplier
         self.max_grad_norm = max_grad_norm
@@ -47,17 +47,46 @@ class DPConfig:
 
 
 class FlowerClient(fl.client.NumPyClient):
+    """
+    Federated learning client for MEDfl real-world pipelines.
+
+    This client:
+      - connects to a Flower server,
+      - loads local CSV data,
+      - applies server-provided schema (features/target),
+      - handles train/val/test splits (fractions or test_ids),
+      - optionally enables differential privacy with Opacus.
+
+    Parameters
+    ----------
+    server_address : str
+        Address of the Flower server, e.g. ``"127.0.0.1:8080"``.
+    data_path : str, optional
+        Path to the local CSV file. Default is ``"data/data.csv"``.
+    dp_config : DPConfig, optional
+        Differential privacy configuration. If ``None``, DP is disabled.
+    val_frac : float, optional
+        Client-side validation fraction override. If ``None``, use server value.
+    test_frac : float, optional
+        Client-side test fraction override. If ``None``, use server value.
+    id_col : str, optional
+        Name of the ID column used when selecting test samples via ``test_ids``.
+    test_ids : str, optional
+        Comma-separated list of IDs (or line numbers) to use as test set.
+    seed : int, optional
+        Random seed used for splits. Default is ``42``.
+    """
     def __init__(
         self,
-        server_address: str,
-        data_path: str = "data/data.csv",
-        dp_config: DPConfig = None,
+        server_address,
+        data_path="data/data.csv",
+        dp_config=None,
         # NEW (optional client overrides; do NOT remove old args)
-        val_frac: float | None = None,
-        test_frac: float | None = None,
-        id_col: str | None = None,
-        test_ids: str | None = None,
-        seed: int = 42,
+        val_frac=None,
+        test_frac=None,
+        id_col=None,
+        test_ids=None,
+        seed=42,
     ):
         self.server_address = server_address
         self.dp_config = dp_config
@@ -83,31 +112,31 @@ class FlowerClient(fl.client.NumPyClient):
         self.y_tensor = torch.tensor(y_default, dtype=torch.float32)
 
         # Placeholders; we lazily build loaders/model on the first fit when we see server config
-        self.train_loader: DataLoader | None = None
-        self.val_loader: DataLoader | None = None
-        self.test_loader: DataLoader | None = None
+        self.train_loader = None
+        self.val_loader = None
+        self.test_loader = None
 
-        self.model: Net | None = None
+        self.model = None
         self.criterion = nn.BCEWithLogitsLoss()
-        self.optimizer: optim.Optimizer | None = None
+        self.optimizer = None
 
         # Effective settings (filled at first fit)
-        self.effective_features: list[str] = self.feature_names[:]
-        self.effective_target: str = self.target_name
-        self.effective_val_frac: float = float(self.client_val_frac) if self.client_val_frac is not None else 0.0
-        self.effective_test_frac: float = float(self.client_test_frac) if self.client_test_frac is not None else 0.0
+        self.effective_features = self.feature_names[:]
+        self.effective_target = self.target_name
+        self.effective_val_frac = float(self.client_val_frac) if self.client_val_frac is not None else 0.0
+        self.effective_test_frac = float(self.client_test_frac) if self.client_test_frac is not None else 0.0
 
         self._initialized = False
         self._dp_attached = False  # to avoid wrapping twice
 
     # ---------- helpers
 
-    def _mk_loader(self, X: np.ndarray, y: np.ndarray, batch_size: int, shuffle: bool) -> DataLoader:
+    def _mk_loader(self, X, y, batch_size, shuffle):
         x_t = torch.tensor(X, dtype=torch.float32)
         y_t = torch.tensor(y, dtype=torch.float32)
         return DataLoader(TensorDataset(x_t, y_t), batch_size=batch_size, shuffle=shuffle)
 
-    def _lazy_init_from_server_config(self, config: dict):
+    def _lazy_init_from_server_config(self, config):
         """
         Build model and (train, val, test) loaders once, using:
           - Server-enforced schema: config['features'] (comma-separated), config['target']
@@ -226,7 +255,7 @@ class FlowerClient(fl.client.NumPyClient):
         # ---------- build loaders
         batch_size = self.dp_config.batch_size if self.dp_config else 32
         self.train_loader = self._mk_loader(X_train, y_train, batch_size, shuffle=True)
-        self.val_loader = self._mk_loader(X_val, y_val, batch_size= batch_size, shuffle=False) if len(y_val) else None
+        self.val_loader = self._mk_loader(X_val, y_val, batch_size=batch_size, shuffle=False) if len(y_val) else None
         self.test_loader = self._mk_loader(X_test, y_test, batch_size, shuffle=False)
 
         # ---------- model/optimizer
@@ -414,7 +443,7 @@ class FlowerClient(fl.client.NumPyClient):
             "gpu_count": gpu_count,
         }
 
-    def start(self) -> None:
+    def start(self):
         fl.client.start_numpy_client(
             server_address=self.server_address,
             client=self,

MEDfl/rw/server.py

@@ -1,7 +1,6 @@
 import flwr as fl
 from flwr.server.strategy import FedAvg
 from flwr.server.server import ServerConfig
-from typing import Optional, Any
 from MEDfl.rw.strategy import Strategy
 import asyncio
 from flwr.server.client_manager import ClientManager
@@ -29,11 +28,11 @@ class FederatedServer:
 
     def __init__(
         self,
-        host: str = "0.0.0.0",
-        port: int = 8080,
-        num_rounds: int = 3,
-        strategy: Optional[Strategy] = None,
-        certificates: Optional[Any] = None,
+        host="0.0.0.0",
+        port=8080,
+        num_rounds=3,
+        strategy=None,
+        certificates=None,
     ):
         """
         Initialize the FederatedServer.
@@ -60,7 +59,7 @@ class FederatedServer:
         self.connected_clients = []  # Track connected client IDs
 
 
-    def start(self) -> None:
+    def start(self):
         """
         Start the Flower server with the configured strategy and track client connections.
         """
@@ -89,7 +88,7 @@ class TrackingClientManager(fl.server.client_manager.SimpleClientManager):
         client_properties (dict): Placeholder for storing client-specific properties.
     """
 
-    def __init__(self, server: FederatedServer):
+    def __init__(self, server):
         """
         Initialize the TrackingClientManager.
 
@@ -100,7 +99,7 @@ class TrackingClientManager(fl.server.client_manager.SimpleClientManager):
         self.server = server
         self.client_properties = {}
 
-    def register(self, client: ClientProxy) -> bool:
+    def register(self, client):
         """
         Register a client and log its connection.
 
@@ -116,7 +115,7 @@ class TrackingClientManager(fl.server.client_manager.SimpleClientManager):
             asyncio.run(self._fetch_and_log_hostname(client))
         return success
 
-    async def _fetch_and_log_hostname(self, client: ClientProxy):
+    async def _fetch_and_log_hostname(self, client):
         """
         Asynchronously fetch and log the client's hostname or CID.
 

MEDfl/rw/strategy.py

@@ -3,7 +3,6 @@
 import os
 import numpy as np
 import flwr as fl
-from typing import Callable, Optional, Dict, Any, List, Tuple
 from flwr.common import GetPropertiesIns
 from flwr.server.client_manager import ClientManager
 from flwr.server.client_proxy import ClientProxy
@@ -14,18 +13,14 @@ import torch
 # ===================================================
 # Custom metric aggregation functions
 # ===================================================
-def aggregate_fit_metrics(
-    results: List[Tuple[int, Dict[str, float]]]
-) -> Dict[str, float]:
+def aggregate_fit_metrics(results):
     total = sum(n for n, _ in results)
     loss = sum(m.get("train_loss", 0.0) * n for n, m in results) / total
     acc = sum(m.get("train_accuracy", 0.0) * n for n, m in results) / total
     auc = sum(m.get("train_auc", 0.0) * n for n, m in results) / total
     return {"train_loss": loss, "train_accuracy": acc, "train_auc": auc}
 
-def aggregate_eval_metrics(
-    results: List[Tuple[int, Dict[str, float]]]
-) -> Dict[str, float]:
+def aggregate_eval_metrics(results):
     total = sum(n for n, _ in results)
     loss = sum(m.get("eval_loss", 0.0) * n for n, m in results) / total
     acc = sum(m.get("eval_accuracy", 0.0) * n for n, m in results) / total
@@ -71,33 +66,33 @@ class Strategy:
 
     def __init__(
         self,
-        name: str = "FedAvg",
-        fraction_fit: float = 1.0,
-        fraction_evaluate: float = 1.0,
-        min_fit_clients: int = 2,
-        min_evaluate_clients: int = 2,
-        min_available_clients: int = 2,
-        initial_parameters: Optional[List[Any]] = None,
-        evaluate_fn: Optional[Callable] = None,
-        fit_metrics_aggregation_fn: Optional[Callable] = None,
-        evaluate_metrics_aggregation_fn: Optional[Callable] = None,
-        local_epochs: int = 1,
-        threshold: float = 0.5,
-        learning_rate: float = 0.01,
-        optimizer_name: str = "SGD",
-        savingPath: str = "",
-        saveOnRounds: int = 3,
-        total_rounds: int = 3,
-        features: str = "",
-        target: str = "",
-        val_fraction: float = 0.10,
-        test_fraction: float = 0.10,
+        name="FedAvg",
+        fraction_fit=1.0,
+        fraction_evaluate=1.0,
+        min_fit_clients=2,
+        min_evaluate_clients=2,
+        min_available_clients=2,
+        initial_parameters=None,
+        evaluate_fn=None,
+        fit_metrics_aggregation_fn=None,
+        evaluate_metrics_aggregation_fn=None,
+        local_epochs=1,
+        threshold=0.5,
+        learning_rate=0.01,
+        optimizer_name="SGD",
+        savingPath="",
+        saveOnRounds=3,
+        total_rounds=3,
+        features="",
+        target="",
+        val_fraction=0.10,
+        test_fraction=0.10,
         # NEW: splitting control (added at the end to not break existing calls)
-        split_mode: str = "global",                  # "global" or "per_client"
-        client_fractions: Optional[Dict[str, Dict[str, Any]]] = None,
+        split_mode="global",                  # "global" or "per_client"
+        client_fractions=None,
         # NEW: id column for test_ids mapping
-        id_col: str = "id",
-    ) -> None:
+        id_col="id",
+    ):
         self.name = name
         self.fraction_fit = fraction_fit
         self.fraction_evaluate = fraction_evaluate
@@ -128,14 +123,14 @@ class Strategy:
         self.client_fractions = client_fractions or {}
         self.id_col = id_col
 
-        self.strategy_object: Optional[fl.server.strategy.Strategy] = None
+        self.strategy_object = None
 
-    def create_strategy(self) -> None:
+    def create_strategy(self):
         # 1) Pick the Flower Strategy class
         StrategyClass = getattr(fl.server.strategy, self.name)
 
         # 2) Define on_fit_config_fn _before_ instantiation (global defaults)
-        def fit_config_fn(server_round: int) -> Dict[str, Any]:
+        def fit_config_fn(server_round):
             return {
                 "local_epochs":   self.local_epochs,
                 "threshold":      self.threshold,
@@ -150,7 +145,7 @@ class Strategy:
             }
 
         # 3) Build params including on_fit_config_fn
-        params: Dict[str, Any] = {
+        params = {
             "fraction_fit":                     self.fraction_fit,
             "fraction_evaluate":                self.fraction_evaluate,
             "min_fit_clients":                  self.min_fit_clients,
@@ -226,9 +221,9 @@ class Strategy:
         original_conf_fit = strat.configure_fit
 
         def wrapped_conf_fit(
-            server_round: int,
+            server_round,
             parameters,
-            client_manager: ClientManager
+            client_manager
         ):
             selected = original_conf_fit(
                 server_round=server_round,

{medfl-2.0.5.dev3.dist-info → medfl-2.0.5.dev6.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
-Name: MEDfl
-Version: 2.0.5.dev3
+Name: medfl
+Version: 2.0.5.dev6
 Summary: Python Open-source package for simulating federated learning and differential privacy
 Home-page: https://github.com/MEDomics-UdeS/MEDfl
 Author: MEDomics consortium
@@ -14,7 +14,7 @@ Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
 Classifier: License :: OSI Approved :: MIT License
 Classifier: Programming Language :: Python :: 3.8
 Classifier: Programming Language :: Python :: 3.9
-Requires-Python: >=3.8,<3.13
+Requires-Python: >=3.8,<3.15
 Description-Content-Type: text/markdown
 License-File: LICENSE
 Requires-Dist: flwr~=1.18.0

{medfl-2.0.5.dev3.dist-info → medfl-2.0.5.dev6.dist-info}/RECORD

@@ -12,7 +12,7 @@ MEDfl/LearningManager/server.py,sha256=oTgW3K1UT6m4SQBk23FIf23km_BDq9vvjeC6OgY8D
 MEDfl/LearningManager/strategy.py,sha256=BHXpwmt7jx07y45YLUs8FZry2gYQbpiV4vNbHhsksQ4,3435
 MEDfl/LearningManager/utils.py,sha256=zk84uKR_iSsoz1uvKD-l4_jJWV3_5pfUWMIeLNcT1mU,9621
 MEDfl/NetManager/__init__.py,sha256=OpgsIiBg7UA6Bfnu_kqGfEPxU8JfpPxSFU98TOeDTP0,273
-MEDfl/NetManager/database_connector.py,sha256=L46vgTuQIzuk4PVWB7Mo1THX4aerIlIlAU5ZTIVGLs8,1553
+MEDfl/NetManager/database_connector.py,sha256=PZEAesMR-k9jSuthx4wOTkMzUWB3INODOrhQgpMO0kY,1554
 MEDfl/NetManager/dataset.py,sha256=HTV0jrJ4Qlhl2aSJzdFU1lkxGBKtmJ390eBpwfKf_4o,2777
 MEDfl/NetManager/flsetup.py,sha256=CVu_TIU7l3G6DDnwtY6JURbhIZk7gKC3unqWnU-YtlM,11434
 MEDfl/NetManager/net_helper.py,sha256=tyfxmpbleSdfPfo2ezKT0VOvZu660v9nhBuHCpl8pG4,6812
@@ -20,17 +20,17 @@ MEDfl/NetManager/net_manager_queries.py,sha256=j-CLQPjtTLyZuFPhIcwJStD7L7xtZpkmk
 MEDfl/NetManager/network.py,sha256=5t705fzWc-BRg-QPAbAcDv5ckDGzsPwj_Q5V0iTgkx0,6829
 MEDfl/NetManager/node.py,sha256=t90QuYZ8M1X_AG1bwTta0CnlOuodqkmpVda2K7NOgHc,6542
 MEDfl/rw/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-MEDfl/rw/client.py,sha256=Ufa-VCbP83adEs7tV1ltDYakBV2Ge5GCPhjIJ7HsuJ8,17804
+MEDfl/rw/client.py,sha256=K0cMd0GuEWX5m11fzgEkdD2JCmY2ZVThSZr0K8dCQXM,18676
 MEDfl/rw/model.py,sha256=IvC5a1ADyn7mKeSBq1Lk_UnkSJOCErPCmpEPSOFxHpk,5343
-MEDfl/rw/server.py,sha256=TmcWUwMJ5BG7owqIVTtTC6w8bR02SESRT9lXh7BqlOg,4986
-MEDfl/rw/strategy.py,sha256=rrRUvuFxL4rdtjN-q-GgB_QQ4ntH9kSyIbTQtiASjVI,13779
+MEDfl/rw/server.py,sha256=JBNSSX8tihrLIjgDTrGfmsXXOq_00gtS0s9qCgbqrHY,4834
+MEDfl/rw/strategy.py,sha256=uQvd9lS_PZc234ghBKvrhwdc7ooztHzewuvwpWzBsJ8,13206
 MEDfl/scripts/__init__.py,sha256=Pq1weevsPaU7MRMHfBYeyT0EOFeWLeVM6Y1DVz6jw1A,48
 MEDfl/scripts/base.py,sha256=QrmG7gkiPYkAy-5tXxJgJmOSLGAKeIVH6i0jq7G9xnA,752
 MEDfl/scripts/create_db.py,sha256=MnFtZkTueRZ-3qXPNX4JsXjOKj-4mlkxoRhSFdRcvJw,3817
 alembic/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 alembic/env.py,sha256=-aSZ6SlJeK1ZeqHgM-54hOi9LhJRFP0SZGjut-JnY-4,1588
-medfl-2.0.5.dev3.dist-info/licenses/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
-medfl-2.0.5.dev3.dist-info/METADATA,sha256=jqEhW5WJalcs9sBxuN27oY5_e1yjwIRUuvY7pzrSpII,6914
-medfl-2.0.5.dev3.dist-info/WHEEL,sha256=lTU6B6eIfYoiQJTZNc-fyaR6BpL6ehTzU3xGYxn2n8k,91
-medfl-2.0.5.dev3.dist-info/top_level.txt,sha256=dIL9X8HOFuaVSzpg40DVveDPrymWRoShHtspH7kkjdI,14
-medfl-2.0.5.dev3.dist-info/RECORD,,
+medfl-2.0.5.dev6.dist-info/licenses/LICENSE,sha256=OXLcl0T2SZ8Pmy2_dmlvKuetivmyPd5m1q-Gyd-zaYY,35149
+medfl-2.0.5.dev6.dist-info/METADATA,sha256=zznydqnDH8kncx_4RvJtfC71PmauDH7GEpDvAzsbND8,6914
+medfl-2.0.5.dev6.dist-info/WHEEL,sha256=lTU6B6eIfYoiQJTZNc-fyaR6BpL6ehTzU3xGYxn2n8k,91
+medfl-2.0.5.dev6.dist-info/top_level.txt,sha256=dIL9X8HOFuaVSzpg40DVveDPrymWRoShHtspH7kkjdI,14
+medfl-2.0.5.dev6.dist-info/RECORD,,