From 9d1035a5e94f5fe731674cc0538b7ff7af1ecfd3 Mon Sep 17 00:00:00 2001 From: zengdun <928255708@qq.com> Date: Fri, 24 Sep 2021 16:15:12 +0800 Subject: [PATCH] fix --- tests/{ => pipeline_test_a}/test_fedavg_client.py | 0 tests/{ => pipeline_test_a}/test_fedavg_server.py | 0 tests/{ => pipeline_test_b}/test_fedasgd_client.py | 2 +- tests/{ => pipeline_test_b}/test_fedasgd_server.py | 3 +-- 4 files changed, 2 insertions(+), 3 deletions(-) rename tests/{ => pipeline_test_a}/test_fedavg_client.py (100%) rename tests/{ => pipeline_test_a}/test_fedavg_server.py (100%) rename tests/{ => pipeline_test_b}/test_fedasgd_client.py (98%) rename tests/{ => pipeline_test_b}/test_fedasgd_server.py (98%) diff --git a/tests/test_fedavg_client.py b/tests/pipeline_test_a/test_fedavg_client.py similarity index 100% rename from tests/test_fedavg_client.py rename to tests/pipeline_test_a/test_fedavg_client.py diff --git a/tests/test_fedavg_server.py b/tests/pipeline_test_a/test_fedavg_server.py similarity index 100% rename from tests/test_fedavg_server.py rename to tests/pipeline_test_a/test_fedavg_server.py diff --git a/tests/test_fedasgd_client.py b/tests/pipeline_test_b/test_fedasgd_client.py similarity index 98% rename from tests/test_fedasgd_client.py rename to tests/pipeline_test_b/test_fedasgd_client.py index f2fa756f..84240b66 100644 --- a/tests/test_fedasgd_client.py +++ b/tests/pipeline_test_b/test_fedasgd_client.py @@ -68,5 +68,5 @@ def setUp(self) -> None: rank=1), ) - def test_fedavg(self): + def test_fedasgd_client(self): self.client.run() diff --git a/tests/test_fedasgd_server.py b/tests/pipeline_test_b/test_fedasgd_server.py similarity index 98% rename from tests/test_fedasgd_server.py rename to tests/pipeline_test_b/test_fedasgd_server.py index 86e328a5..d53e468a 100644 --- a/tests/test_fedasgd_server.py +++ b/tests/pipeline_test_b/test_fedasgd_server.py @@ -54,8 +54,7 @@ def setUp(self) -> None: world_size=world_size, rank=1), ) - self.client.start() - def test_fedavg(self): + def test_fedasgd_server(self): self.server.run()