Skip to content

Commit aa655d7

Browse files
jopemachineseedspirit
authored andcommitted
fix(BA-1921): status_data not initialized properly when creating single node session (#5217)
1 parent 0b2d88f commit aa655d7

File tree

2 files changed

+9
-2
lines changed

2 files changed

+9
-2
lines changed

changes/5217.fix.md

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1 @@
1+
`status_data` not initialized properly when creating single node session

src/ai/backend/manager/repositories/schedule/repository.py

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -505,13 +505,19 @@ async def finalize_single_node_session(
505505

506506
for kernel in session_row.kernels:
507507
kernel.set_status(
508-
KernelStatus.SCHEDULED, status_info="scheduled", status_changed_at=now
508+
KernelStatus.SCHEDULED,
509+
status_info="scheduled",
510+
status_data={},
511+
status_changed_at=now,
509512
)
510513
kernel.agent = agent_alloc_ctx.agent_id
511514
kernel.agent_addr = agent_alloc_ctx.agent_addr
512515
kernel.scaling_group = sgroup_name
513516
session_row.set_status(
514-
SessionStatus.SCHEDULED, status_info="scheduled", status_changed_at=now
517+
SessionStatus.SCHEDULED,
518+
status_info="scheduled",
519+
status_data={},
520+
status_changed_at=now,
515521
)
516522
if agent_alloc_ctx.agent_id is not None:
517523
agent_ids.append(agent_alloc_ctx.agent_id)

0 commit comments

Comments
 (0)