Skip to content

Commit 4204990

Browse files
author
Andrei Neagu
committed
fixed broken tests
1 parent d6b89a9 commit 4204990

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

services/director-v2/tests/unit/with_dbs/comp_scheduler/test_scheduler_dask.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -255,6 +255,7 @@ async def _return_tasks_pending(job_ids: list[str]) -> list[DaskClientTaskState]
255255
callback=mock.ANY,
256256
metadata=mock.ANY,
257257
hardware_info=mock.ANY,
258+
run_id=mock.ANY,
258259
)
259260
for p in expected_pending_tasks
260261
],
@@ -654,6 +655,7 @@ async def _return_random_task_result(job_id) -> TaskOutputData:
654655
callback=mock.ANY,
655656
metadata=mock.ANY,
656657
hardware_info=mock.ANY,
658+
run_id=mock.ANY,
657659
)
658660
mocked_dask_client.send_computation_tasks.reset_mock()
659661
mocked_dask_client.get_tasks_status.assert_has_calls(

0 commit comments

Comments
 (0)