FedModelKit 0.6.5__tar.gz → 0.6.7__tar.gz

{fedmodelkit-0.6.5 → fedmodelkit-0.6.7}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: FedModelKit
-Version: 0.6.5
+Version: 0.6.7
 Summary: LOW_LEVEL_API: This package contains the core components and protocols for creating, managing, and registering federated learning models using MLflow. It provides utilities for defining local learners, aggregation strategies, and integrating them with MLflow for tracking and deployment.
 Author-email: ceresale <alessandro.ceresi@upm.es>
 License-File: LICENSE

{fedmodelkit-0.6.5 → fedmodelkit-0.6.7}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "FedModelKit"
-version = "0.6.5"
+version = "0.6.7"
 description = "LOW_LEVEL_API: This package contains the core components and protocols for creating, managing, and registering federated learning models using MLflow. It provides utilities for defining local learners, aggregation strategies, and integrating them with MLflow for tracking and deployment."
 readme = "README.md"
 authors = [

{fedmodelkit-0.6.5 → fedmodelkit-0.6.7}/src/FedModelKit/templates/client_app_template.py

@@ -35,7 +35,7 @@ def query(msg: Message, ctx: Context) -> Message:
     # To store in context other objects, you can use ctx.state.<object_name> = <object>
     ctx.state.local_learner = federated_model
 
-    return msg.create_reply(RecordDict())
+    return Message(RecordDict(), reply_to=msg)
 
 @app.train()
 def train(msg: Message, ctx: Context):
@@ -71,7 +71,7 @@ def train(msg: Message, ctx: Context):
     ctx.state.local_learner =  local_learner
 
     # Return the reply message to the server
-    return msg.create_reply(reply_content)
+    return Message(reply_content, reply_to=msg)
 
 @app.evaluate()
 def eval(msg: Message, ctx: Context):
@@ -103,4 +103,4 @@ def eval(msg: Message, ctx: Context):
     ctx.state.local_learner =  local_learner
 
     # Return the reply message to the server
-    return msg.create_reply(reply_content)
+    return Message(reply_content, reply_to=msg)

{fedmodelkit-0.6.5 → fedmodelkit-0.6.7}/src/FedModelKit/templates/extern_pyproject_template.toml

@@ -15,4 +15,5 @@ dependencies = [
     "syft-flwr>=0.1.5",
     "torch==2.7.0",
     "FedModelKit>=0.6.1",
+    "ray>=2.47.1",
 ]

{fedmodelkit-0.6.5 → fedmodelkit-0.6.7}/src/FedModelKit/templates/server_app_template.py

@@ -26,7 +26,7 @@ app = fl.server.ServerApp()
 
 
 @app.main()
-def main(driver: Grid, context: Context):
+def main(grid: Grid, context: Context):
     """
     Main function to run the federated learning server.
 
@@ -41,7 +41,7 @@ def main(driver: Grid, context: Context):
     print("Starting test run")
 
     # Get node IDs of connected clients
-    node_ids = driver.get_node_ids()
+    node_ids = grid.get_node_ids()
 
     # Initialize the federated model
     global_model = Net()
@@ -51,14 +51,14 @@ def main(driver: Grid, context: Context):
     messages = []
     for idx, node_id in enumerate(node_ids):
         # Create messages to send to clients
-        RecordDict = RecordDict()
+        record_dict = RecordDict()
 
         # Add a config with information to send the client for the query
-        RecordDict.config_records["fancy_config"] = ConfigRecord({"num_clients": len(node_ids), "client_id": idx})
+        record_dict.config_records["fancy_config"] = ConfigRecord({"num_clients": len(node_ids), "client_id": idx})
 
         # Create a query message for each client
-        message = driver.create_message(
-            content=RecordDict,
+        message = Message(
+            content=record_dict,
             message_type=MessageType.QUERY,
             dst_node_id=node_id,
             group_id=str(1),
@@ -67,7 +67,7 @@ def main(driver: Grid, context: Context):
         messages.append(message)
 
     # Send training messages to clients
-    all_replies = list(driver.send_and_receive(messages))
+    all_replies = list(grid.send_and_receive(messages))
     print(f"Received {len(all_replies)} answers")
     
 
@@ -78,16 +78,16 @@ def main(driver: Grid, context: Context):
         messages = []
         for idx, node_id in enumerate(node_ids):
             # Create messages to send to clients
-            RecordDict = RecordDict()
+            record_dict = RecordDict()
 
             # Add model parameters to record
-            RecordDict.array_records["fancy_model"] = global_model.get_parameters()
+            record_dict.array_records["fancy_model"] = global_model.get_parameters()
             # Add a config with information to send the client for training
-            RecordDict.config_records["fancy_config"] = ConfigRecord({"local_epochs": 3})
+            record_dict.config_records["fancy_config"] = ConfigRecord({"local_epochs": 3})
 
             # Create a training message for each client
-            message = driver.create_message(
-                content=RecordDict,
+            message = Message(
+                content=record_dict,
                 message_type=MessageType.TRAIN,
                 dst_node_id=node_id,
                 group_id=str(server_round),
@@ -96,7 +96,7 @@ def main(driver: Grid, context: Context):
             messages.append(message)
 
         # Send training messages to clients
-        all_replies = list(driver.send_and_receive(messages))
+        all_replies = list(grid.send_and_receive(messages))
         print(f"Received {len(all_replies)} results")
 
         # Print metrics received from clients
@@ -112,16 +112,16 @@ def main(driver: Grid, context: Context):
         messages = []
         for idx, node_id in enumerate(node_ids):
             # Create evaluation messages for clients
-            RecordDict = RecordDict()
+            record_dict = RecordDict()
 
             # Add updated model parameters to record
-            RecordDict.array_records["fancy_model"] = new_array_records
+            record_dict.array_records["fancy_model"] = new_array_records
             # Add a config with information to send the client for evaluation
-            RecordDict.config_records["fancy_config"] = ConfigRecord({"local_epochs": 3})
+            record_dict.config_records["fancy_config"] = ConfigRecord({"local_epochs": 3})
 
             # Create an evaluation message for each client
-            message = driver.create_message(
-                content=RecordDict,
+            message = Message(
+                content=record_dict,
                 message_type=MessageType.EVALUATE,
                 dst_node_id=node_id,
                 group_id=str(server_round),
@@ -130,7 +130,7 @@ def main(driver: Grid, context: Context):
             messages.append(message)
 
         # Send evaluation messages to clients
-        all_replies = list(driver.send_and_receive(messages))
+        all_replies = list(grid.send_and_receive(messages))
         print(f"Received {len(all_replies)} results")
 
         # Print evaluation metrics received from clients

{fedmodelkit-0.6.5 → fedmodelkit-0.6.7}/uv.lock

@@ -476,7 +476,7 @@ wheels = [
 
 [[package]]
 name = "fedmodelkit"
-version = "0.5.0"
+version = "0.6.6"
 source = { editable = "." }
 dependencies = [
     { name = "flwr" },