[refactor] fix test.

SMILELab-FL · Apr 22, 2022 · 8bfa207 · 8bfa207
1 parent 766dc4f
commit 8bfa207
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/test_core/test_server/test_handler.py b/tests/test_core/test_server/test_handler.py
@@ -44,7 +44,7 @@ def test_update_model(self):
         coming_parameters = SerializationTool.serialize_model(coming_model)
 
         self.AsyncHandler._update_global_model(payload=[coming_parameters,
-                                        random.randint(1, 10)])
+                                        torch.Tensor([random.randint(1, 10)])])
 
         parameter_list = []
         for id in range(self.SyncHandler.client_num_per_round):