fix controller tests for different versions

wandb · May 2, 2024 · 2d2b25e · 2d2b25e
1 parent df4143b
commit 2d2b25e
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tests/pytest_tests/unit_tests/test_launch/test_controller/test_local_process_controller.py b/tests/pytest_tests/unit_tests/test_launch/test_controller/test_local_process_controller.py
@@ -66,7 +66,7 @@ async def test_reconcile_clear_unowned_item(mocked_test_manager_reconile):
  mocked_test_manager_reconile.active_runs = {"not-test-id": MagicMock()}
  mocked_test_manager_reconile.release_item = MagicMock()
  await mocked_test_manager_reconile.reconcile()
- assert mocked_test_manager_reconile.release_item.called_once_with("not-test-id")
+ mocked_test_manager_reconile.release_item.assert_called_once_with("not-test-id")
 
 
 @pytest.fixture

diff --git a/tests/pytest_tests/unit_tests/test_launch/test_controller/test_scheduler_controller.py b/tests/pytest_tests/unit_tests/test_launch/test_controller/test_scheduler_controller.py
@@ -24,7 +24,7 @@ def scheduler_controller(controller_config, jobset):
 
 
 @pytest.mark.asyncio
-async def test_scheduler_controller_poll(scheduler_manager):
+async def test_scheduler_controller_poll(scheduler_controller):
  await scheduler_controller.poll()
  assert scheduler_controller._scheduler_jobs_queue.get.called_once()
  assert scheduler_controller._controller.launch_scheduler_item.called_once()