rename function

lablup · Jun 23, 2024 · 3c0ab97 · 3c0ab97
1 parent 1151274
commit 3c0ab97
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/ai/backend/agent/agent.py b/src/ai/backend/agent/agent.py
@@ -1679,7 +1679,7 @@ async def execute_batch(
                 SessionFailureEvent(session_id, KernelLifecycleEventReason.TASK_CANCELLED, -2),
             )
 
-    async def trigger_batch_execution(
+    async def create_batch_execution_task(
         self,
         session_id: SessionId,
         kernel_id: KernelId,

diff --git a/src/ai/backend/agent/server.py b/src/ai/backend/agent/server.py
@@ -635,7 +635,7 @@ async def trigger_batch_execution(
         log.info(
             "rpc::trigger_batch_execution(k:{0}, s:{1}, code:{2})", kernel_id, session_id, code
         )
-        await self.agent.trigger_batch_execution(
+        await self.agent.create_batch_execution_task(
             SessionId(UUID(session_id)), KernelId(UUID(kernel_id)), code
         )