Skip to content

Commit

Permalink
update names
Browse files Browse the repository at this point in the history
  • Loading branch information
panh99 committed Jan 22, 2024
1 parent 4b238b1 commit c254177
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions src/py/flwr/driver/driver_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -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
)
Expand Down

0 comments on commit c254177

Please sign in to comment.