change field names

lablup · Aug 1, 2024 · 7092cbb · 7092cbb
1 parent 48ccec7
commit 7092cbb
Showing 1 changed file with 7 additions and 3 deletions.
diff --git a/src/ai/backend/manager/scheduler/dispatcher.py b/src/ai/backend/manager/scheduler/dispatcher.py
@@ -57,6 +57,7 @@
 from ai.backend.common.plugin.hook import PASSED, HookResult
 from ai.backend.common.types import (
     AgentId,
+    AgentKernelRegistryByStatus,
     ClusterMode,
     RedisConnectionInfo,
     ResourceSlot,
@@ -75,7 +76,7 @@
     SessionNotFound,
 )
 from ..defs import SERVICE_MAX_RETRIES, LockID
-from ..exceptions import convert_to_status_data
+from ..exceptions import MultiAgentError, convert_to_status_data
 from ..models import (
     AgentRow,
     AgentStatus,
@@ -315,10 +316,13 @@ def _pipeline(r: Redis) -> RedisPipeline:
                                 sgroup_name,
                             )
                         except Exception as e:
-                            log.exception("schedule({}): scheduling error!\n{}", sgroup_name, repr(e))
+                            log.exception(
+                                "schedule({}): scheduling error!\n{}", sgroup_name, repr(e)
+                            )
                         else:
                             if (
-                                AgentResourceSyncTrigger.AFTER_SCHEDULING in agent_resource_sync_trigger
+                                AgentResourceSyncTrigger.AFTER_SCHEDULING
+                                in agent_resource_sync_trigger
                                 and kernel_agent_bindings
                             ):
                                 selected_agent_ids = [