From c254177026a8c61d85055d325caa16a63beac29b Mon Sep 17 00:00:00 2001 From: Heng Pan Date: Mon, 22 Jan 2024 17:01:44 +0000 Subject: [PATCH] update names --- src/py/flwr/driver/driver_test.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/src/py/flwr/driver/driver_test.py b/src/py/flwr/driver/driver_test.py index 81deb5d35ac6..e48b9562f36b 100644 --- a/src/py/flwr/driver/driver_test.py +++ b/src/py/flwr/driver/driver_test.py @@ -170,9 +170,9 @@ class TestDriverRetryMechanism(unittest.TestCase): def setUp(self) -> None: """Initialize mock GrpcDriver and Driver instance before each test.""" mock_response = Mock() - mock_response.workload_id = 61016 + mock_response.run_id = 61016 self.mock_grpc_driver = Mock() - self.mock_grpc_driver.create_workload.return_value = mock_response + self.mock_grpc_driver.create_run.return_value = mock_response self.patcher = patch( "flwr.driver.driver.GrpcDriver", return_value=self.mock_grpc_driver )