From 992aac6819280dd0bfda9a6c93afde43714ea4b7 Mon Sep 17 00:00:00 2001 From: Qing Wang Date: Tue, 11 Jul 2023 11:08:24 +0800 Subject: [PATCH] revert Signed-off-by: Qing Wang --- tests/test_fed_get.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/tests/test_fed_get.py b/tests/test_fed_get.py index 714ef7c..f49fc6a 100644 --- a/tests/test_fed_get.py +++ b/tests/test_fed_get.py @@ -55,7 +55,7 @@ def run(party): fed.init(cluster=cluster, party=party) epochs = 3 - alice_model = MyModel.remote("alice", 2) + alice_model = MyModel.party("alice").remote("alice", 2) bob_model = MyModel.party("bob").remote("bob", 4) all_mean_weights = []