blazefl 2.0.0.dev7__tar.gz → 2.0.0.dev8__tar.gz

{blazefl-2.0.0.dev7 → blazefl-2.0.0.dev8}/.github/workflows/ci.yaml

@@ -11,7 +11,6 @@ on:
   push:
     branches:
       - main
-      - develop
 
 jobs:
   run:
@@ -39,4 +38,14 @@ jobs:
         run: uv run mypy src
 
       - name: Run pytest
-        run: uv run pytest tests
+        run: uv run pytest tests --junitxml=pytest.xml
+        continue-on-error: true
+
+      - name: Upload pytest results
+        if: ${{ always() }}
+        uses: actions/upload-artifact@v4
+        with:
+          name: pytest-results
+          path: pytest.xml
+
+

{blazefl-2.0.0.dev7 → blazefl-2.0.0.dev8}/.github/workflows/publish.yaml

@@ -4,8 +4,8 @@ on:
   release:
     types: ["published"]
   push:
-    branches:
-      - develop
+    tags:
+      - 'v*.*.*.dev*'
 
 jobs:
   run:

{blazefl-2.0.0.dev7 → blazefl-2.0.0.dev8}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: blazefl
-Version: 2.0.0.dev7
+Version: 2.0.0.dev8
 Summary: A blazing-fast and lightweight simulation framework for Federated Learning.
 Author-email: kitsuyaazuma <kitsuyaazuma@gmail.com>
 License-File: LICENSE

{blazefl-2.0.0.dev7 → blazefl-2.0.0.dev8}/examples/experimental-freethreaded/pyproject.toml

@@ -5,7 +5,7 @@ description = "Add your description here"
 readme = "README.md"
 requires-python = ">=3.13"
 dependencies = [
-    "blazefl>=2.0.0dev6",
+    "blazefl>=2.0.0dev7",
     "hydra-core>=1.3.2",
     "torch>=2.7.1",
     "torchvision>=0.22.1",

{blazefl-2.0.0.dev7 → blazefl-2.0.0.dev8}/examples/experimental-freethreaded/uv.lock

@@ -10,16 +10,16 @@ sdist = { url = "https://files.pythonhosted.org/packages/3e/38/7859ff46355f76f8d
 
 [[package]]
 name = "blazefl"
-version = "2.0.0.dev6"
+version = "2.0.0.dev7"
 source = { registry = "https://pypi.org/simple" }
 dependencies = [
     { name = "numpy" },
     { name = "torch" },
     { name = "tqdm" },
 ]
-sdist = { url = "https://files.pythonhosted.org/packages/d7/76/d247942d051447dc1c6dbc25b2d55b4cdf146cc0ebd3594c0c33628f465c/blazefl-2.0.0.dev6.tar.gz", hash = "sha256:d4875f03872917dfd99b3a70ac2c6efc3483ee0e42e484d7328e0810fe5141b9", size = 615380, upload_time = "2025-06-16T16:26:50.929Z" }
+sdist = { url = "https://files.pythonhosted.org/packages/ae/30/2dc1c42d2b1c2ff7742fa37e77a4c08e0afe2a1170078a3364d5e9bce153/blazefl-2.0.0.dev7.tar.gz", hash = "sha256:8b77dc2c3c28ced6ad3c9662ff1eb8ca65175097e68aa97ebd4a74d114088bd9", size = 615335, upload_time = "2025-06-19T14:39:33.54Z" }
 wheels = [
-    { url = "https://files.pythonhosted.org/packages/c8/30/eba32caaed89d04c925f676d48fb50bad7bed3ce2c86c6ebcaf2eb92aff1/blazefl-2.0.0.dev6-py3-none-any.whl", hash = "sha256:e38c3739be9ef2bc7bb8ba213da3319e4be8f9ea9f27ecd661cbc70c0cedd655", size = 26215, upload_time = "2025-06-16T16:26:49.512Z" },
+    { url = "https://files.pythonhosted.org/packages/6c/b8/92d9d6b6b9e2e128b7d532ea0a96032adf3da1122e8f35089589368c4484/blazefl-2.0.0.dev7-py3-none-any.whl", hash = "sha256:aa1fe627b6526d7333c33d3b860ca68b796715d8c90aa8387d0f480a67bd64d1", size = 26390, upload_time = "2025-06-19T14:39:32.293Z" },
 ]
 
 [[package]]
@@ -50,7 +50,7 @@ dev = [
 
 [package.metadata]
 requires-dist = [
-    { name = "blazefl", specifier = ">=2.0.0.dev6" },
+    { name = "blazefl", specifier = ">=2.0.0.dev7" },
     { name = "hydra-core", specifier = ">=1.3.2" },
     { name = "torch", specifier = ">=2.7.1" },
     { name = "torchvision", specifier = ">=0.22.1" },

{blazefl-2.0.0.dev7 → blazefl-2.0.0.dev8}/pyproject.toml

@@ -1,6 +1,6 @@
 [project]
 name = "blazefl"
-version = "2.0.0.dev7"
+version = "2.0.0.dev8"
 description = "A blazing-fast and lightweight simulation framework for Federated Learning."
 readme = "README.md"
 authors = [

{blazefl-2.0.0.dev7 → blazefl-2.0.0.dev8}/src/blazefl/contrib/fedavg.py

@@ -1,6 +1,5 @@
 import random
 import threading
-from copy import deepcopy
 from dataclasses import dataclass
 from enum import StrEnum
 from pathlib import Path
@@ -195,13 +194,11 @@ class FedAvgBaseServerHandler(
         Returns:
             torch.Tensor: Aggregated model parameters.
         """
-        parameters = torch.stack(parameters_list, dim=-1)
-        weights = torch.tensor(weights_list)
-        weights = weights / torch.sum(weights)
-
-        serialized_parameters = torch.sum(parameters * weights, dim=-1)
-
-        return serialized_parameters
+        total_weight = sum(weights_list)
+        aggregated_parameters = parameters_list[0].clone().zero_()
+        for parameters, weight in zip(parameters_list, weights_list, strict=True):
+            aggregated_parameters.add_(parameters, alpha=weight / total_weight)
+        return aggregated_parameters
 
     @staticmethod
     def evaluate(
@@ -433,7 +430,7 @@ class FedAvgBaseClientTrainer(
         Returns:
             list[FedAvgUplinkPackage]: A list of uplink packages.
         """
-        package = deepcopy(self.cache)
+        package = self.cache
         self.cache = []
         return package
 
@@ -728,7 +725,7 @@ class FedAvgProcessPoolClientTrainer(
         Returns:
             list[FedAvgUplinkPackage]: A list of uplink packages.
         """
-        package = deepcopy(self.cache)
+        package = self.cache
         self.cache = []
         return package
 
@@ -845,6 +842,6 @@ class FedAvgThreadPoolClientTrainer(
         return FedAvgUplinkPackage(model_parameters, data_size)
 
     def uplink_package(self) -> list[FedAvgUplinkPackage]:
-        package = deepcopy(self.cache)
+        package = self.cache
         self.cache = []
         return package

{blazefl-2.0.0.dev7 → blazefl-2.0.0.dev8}/tests/test_contrib/test_fedavg.py

@@ -262,7 +262,7 @@ def test_base_handler_and_process_pool_trainer_integration_keyboard_interrupt(
 
     os.kill(proc.pid, signal.SIGINT)
 
-    proc.join(timeout=5)
+    proc.join(timeout=10)
     assert not proc.is_alive()
 
 

{blazefl-2.0.0.dev7 → blazefl-2.0.0.dev8}/uv.lock

@@ -82,7 +82,7 @@ wheels = [
 
 [[package]]
 name = "blazefl"
-version = "2.0.0.dev7"
+version = "2.0.0.dev8"
 source = { editable = "." }
 dependencies = [
     { name = "numpy" },