Skip to content

Commit 8bfa207

Browse files
committed
[refactor] fix test.
1 parent 766dc4f commit 8bfa207

1 file changed

Lines changed: 1 addition & 1 deletion

File tree

tests/test_core/test_server/test_handler.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -44,7 +44,7 @@ def test_update_model(self):
4444
coming_parameters = SerializationTool.serialize_model(coming_model)
4545

4646
self.AsyncHandler._update_global_model(payload=[coming_parameters,
47-
random.randint(1, 10)])
47+
torch.Tensor([random.randint(1, 10)])])
4848

4949
parameter_list = []
5050
for id in range(self.SyncHandler.client_num_per_round):

0 commit comments

Comments
 (0)