diff --git a/examples/mnist/run_experiment.py b/examples/mnist/run_experiment.py index a710552..cd512a2 100644 --- a/examples/mnist/run_experiment.py +++ b/examples/mnist/run_experiment.py @@ -10,7 +10,7 @@ HTTPClient, HTTPServer, ModelManager, - run_coordinator, + coordinate, ) from nanofed.data import load_mnist_data from nanofed.models import MNISTModel @@ -113,7 +113,7 @@ async def main(): # Run the coordinator and clients concurrently await asyncio.gather( - run_coordinator(coordinator), + coordinate(coordinator), run_client("client_1", "http://0.0.0.0:8080", data_dir), run_client("client_2", "http://0.0.0.0:8080", data_dir), run_client("client_3", "http://0.0.0.0:8080", data_dir), diff --git a/nanofed/__init__.py b/nanofed/__init__.py index a2179fc..22e016e 100644 --- a/nanofed/__init__.py +++ b/nanofed/__init__.py @@ -4,7 +4,7 @@ from nanofed.orchestration import ( Coordinator, CoordinatorConfig, - run_coordinator, + coordinate, ) from nanofed.server import FedAvgAggregator, ModelManager from nanofed.trainer import TorchTrainer @@ -17,7 +17,7 @@ "CoordinatorConfig", "FedAvgAggregator", "ModelManager", - "run_coordinator", + "coordinate", ] diff --git a/nanofed/orchestration/__init__.py b/nanofed/orchestration/__init__.py index c33f105..2b1a895 100644 --- a/nanofed/orchestration/__init__.py +++ b/nanofed/orchestration/__init__.py @@ -1,6 +1,6 @@ from .coordinator import Coordinator, CoordinatorConfig from .types import ClientInfo, RoundMetrics, RoundStatus, TrainingProgress -from .utils import run_coordinator +from .utils import coordinate __all__ = [ "Coordinator", @@ -9,5 +9,5 @@ "RoundMetrics", "RoundStatus", "TrainingProgress", - "run_coordinator", + "coordinate", ] diff --git a/nanofed/orchestration/utils.py b/nanofed/orchestration/utils.py index d176124..09fe5f1 100644 --- a/nanofed/orchestration/utils.py +++ b/nanofed/orchestration/utils.py @@ -2,7 +2,7 @@ from nanofed.utils import Logger -async def run_coordinator(coordinator: Coordinator) -> None: +async def coordinate(coordinator: Coordinator) -> None: """Run the coordinator and consume training metrics. This function continuously starts training rounds and processes