@@ -1150,8 +1150,8 @@ async def get_terminating_sessions_by_ids(
11501150 terminating_sessions .append (
11511151 TerminatingSessionData (
11521152 session_id = session_row .id ,
1153- main_access_key = AccessKey (session_row . access_key )
1154- if session_row . access_key
1153+ main_access_key = AccessKey (owner_main_ak )
1154+ if owner_main_ak
11551155 else AccessKey ("" ),
11561156 creation_id = session_row .creation_id or "" ,
11571157 status = session_row .status ,
@@ -1864,7 +1864,7 @@ async def allocate_sessions(
18641864 ScheduledSessionData (
18651865 session_id = allocation .session_id ,
18661866 creation_id = creation_id ,
1867- main_access_key = access_key ,
1867+ main_access_key = main_access_key ,
18681868 reason = "triggered-by-scheduler" ,
18691869 )
18701870 )
@@ -2924,7 +2924,9 @@ async def _get_sessions_by_statuses(
29242924 scheduled_session = ScheduledSessionData (
29252925 session_id = session .id ,
29262926 creation_id = session .creation_id or "" ,
2927- main_access_key = AccessKey (session .access_key ) if session .access_key else AccessKey ("" ),
2927+ main_access_key = AccessKey (session .access_key )
2928+ if session .access_key
2929+ else AccessKey ("" ),
29282930 reason = "triggered-by-scheduler" ,
29292931 )
29302932 scheduled_sessions .append (scheduled_session )
0 commit comments