diff --git a/src/ai/backend/manager/api/session.py b/src/ai/backend/manager/api/session.py index 032be7457d..ac033ff0b2 100644 --- a/src/ai/backend/manager/api/session.py +++ b/src/ai/backend/manager/api/session.py @@ -2149,9 +2149,11 @@ async def get_container_logs( session_name, owner_access_key, allow_stale=True, - kernel_loading_strategy=KernelLoadingStrategy.MAIN_KERNEL_ONLY - if kernel_id is None - else KernelLoadingStrategy.ALL_KERNELS, + kernel_loading_strategy=( + KernelLoadingStrategy.MAIN_KERNEL_ONLY + if kernel_id is None + else KernelLoadingStrategy.ALL_KERNELS + ), ) if compute_session.status in DEAD_SESSION_STATUSES: diff --git a/src/ai/backend/manager/models/group.py b/src/ai/backend/manager/models/group.py index a80b8b2f00..7fff185ce2 100644 --- a/src/ai/backend/manager/models/group.py +++ b/src/ai/backend/manager/models/group.py @@ -298,9 +298,11 @@ def from_row(cls, graph_ctx: GraphQueryContext, row: Row) -> Optional[Group]: created_at=row["created_at"], modified_at=row["modified_at"], domain_name=row["domain_name"], - total_resource_slots=row["total_resource_slots"].to_json() - if row["total_resource_slots"] is not None - else {}, + total_resource_slots=( + row["total_resource_slots"].to_json() + if row["total_resource_slots"] is not None + else {} + ), allowed_vfolder_hosts=row["allowed_vfolder_hosts"].to_json(), integration_id=row["integration_id"], resource_policy=row["resource_policy"], @@ -751,7 +753,7 @@ async def delete_sessions( """ from .session import SessionRow - stmt = sa.delet(SessionRow).where(SessionRow.group_id == group_id) + stmt = sa.delete(SessionRow).where(SessionRow.group_id == group_id) await db_conn.execute(stmt) @classmethod