Fixed a bug in the many example

Author: frankcorneliusmartin

v6-session-basics/central.py

@@ -66,3 +66,35 @@ def global_sum_dev(client: AlgorithmClient, column: str) -> dict:
         global_sum += output["sum"]
 
     return {"global_sum": global_sum}
+
+
+@central
+@algorithm_client
+def global_sum_dev_many(client: AlgorithmClient, column: str) -> dict:
+    info("Central function that sums the results of the federated sum function")
+
+    # Collect all organization that participate in this collaboration.
+    # These organizations will receive the task to compute the partial.
+    info("Collecting participating organizations")
+    organizations = client.organization.list()
+    ids = [organization.get("id") for organization in organizations]
+
+    info(f"Sending task to {len(ids)} organizations")
+    task = client.task.create(
+        name="central-sum",
+        description="subtask",
+        method="sum_many",
+        organizations=ids,
+        input_={"args": [column], "kwargs": {}},
+    )
+
+    info("Waiting for results...")
+    results = client.wait_for_results(task_id=task.get("id"))
+    info("Partial results are in!")
+
+    info("Computing global sum")
+    global_sums = {}
+    for output in results:
+        global_sums.update(output["sums"])
+
+    return {"global_sum": global_sum}

v6-session-basics/partial.py

@@ -60,5 +60,5 @@ def sum_dev(df1: pd.DataFrame, column: str) -> dict:
 def sum_many(dfs: dict[str, pd.DataFrame], column: str) -> dict:
     sums = {}
     for df_name, df in dfs.items():
-        sums[df_name] = df[column].sum()
+        sums[df_name] = int(df[column].sum())
     return {"sums": sums}