Fix CI

lablup · Oct 16, 2023 · 920231a · 920231a
1 parent 110273e
commit 920231a
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/tests/manager/test_scheduler.py b/tests/manager/test_scheduler.py
@@ -26,6 +26,7 @@
     SessionId,
     SessionTypes,
 )
+from ai.backend.manager.api.context import RaftClusterContext
 from ai.backend.manager.defs import DEFAULT_ROLE
 from ai.backend.manager.models.agent import AgentRow
 from ai.backend.manager.models.image import ImageRow
@@ -1169,13 +1170,15 @@ async def test_manually_assign_agent_available(
     candidate_agents = example_agents
     example_pending_sessions[0].kernels[0].agent = example_agents[0].id
     sess_ctx = example_pending_sessions[0]
+    raft_ctx = RaftClusterContext()
 
     dispatcher = SchedulerDispatcher(
         local_config=mock_local_config,
         shared_config=mock_shared_config,
         event_dispatcher=mock_event_dispatcher,
         event_producer=mock_event_producer,
         lock_factory=file_lock_factory,
+        raft_ctx=raft_ctx,
         registry=registry,
     )