Skip to content
This repository was archived by the owner on Aug 2, 2023. It is now read-only.

Commit 4b5c9cd

Browse files
committed
fix:change typos in session.py that do not match code style
1 parent 38c3f51 commit 4b5c9cd

File tree

1 file changed

+8
-2
lines changed

1 file changed

+8
-2
lines changed

src/ai/backend/manager/api/session.py

Lines changed: 8 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -511,7 +511,8 @@ async def _create(request: web.Request, params: Any) -> web.Response:
511511
resp['sessionId'] = str(kernel_id) # changed since API v5
512512

513513
async def kernelpullprogress(reporter: ProgressReporter) -> None:
514-
progress = [0,0]
514+
progress = [0, 0]
515+
515516
async def _get_status(kernel_id):
516517
async with root_ctx.db.begin() as conn:
517518
query = (
@@ -523,13 +524,15 @@ async def _get_status(kernel_id):
523524
.where(kernels.c.id == kernel_id)
524525
)
525526
result = await conn.execute(query)
527+
526528
return result.first()
527529

528530
async def _update_progress(
529531
app: web.Application,
530532
source: AgentId,
531533
event: KernelPullProgressEvent
532534
) -> None:
535+
# update both current and total
533536
progress[0] = int(event.current_progress)
534537
progress[1] = int(event.total_progress)
535538

@@ -548,7 +551,10 @@ async def _update_progress(
548551
await reporter.update(0)
549552
await asyncio.sleep(0.5)
550553

551-
task_id = await root_ctx.background_task_manager.start(kernelpullprogress, name='kernel_pull_progress')
554+
task_id = await root_ctx.background_task_manager.start(
555+
kernelpullprogress,
556+
name='kernel-pull-progress'
557+
)
552558
resp['background_task'] = str(task_id)
553559
resp['sessionName'] = str(params['session_name'])
554560
resp['status'] = 'PENDING'

0 commit comments

Comments
 (0)