Skip to content

Commit

Permalink
naming update
Browse files Browse the repository at this point in the history
  • Loading branch information
fregataa committed Jul 4, 2024
1 parent feee137 commit 1e4c934
Show file tree
Hide file tree
Showing 2 changed files with 4 additions and 4 deletions.
6 changes: 3 additions & 3 deletions src/ai/backend/manager/models/user.py
Original file line number Diff line number Diff line change
Expand Up @@ -959,7 +959,7 @@ class Arguments:
msg = graphene.String()
bgtask_id = graphene.UUID(
required=False,
description="Added in 24.03.7. ID of backgroun task deleting user owned vfolders.",
description="Added in 24.03.7. ID of background task deleting user owned vfolders.",
)

@classmethod
Expand All @@ -976,7 +976,7 @@ async def mutate(
delete_kernels_by_access_key,
delete_sessions_by_access_key,
)
from .vfolder import VFolderDeletionInfo, delete_vfolders
from .vfolder import VFolderDeletionInfo, delete_vfolders_in_bgtask

graph_ctx: GraphQueryContext = info.context
delete_query = sa.delete(users).where(users.c.email == email)
Expand Down Expand Up @@ -1048,7 +1048,7 @@ async def _pre_purge() -> list[VFolderRow]:
background_task_manager = graph_ctx.background_task_manager

async def _delete_vfolders_task(reporter: ProgressReporter) -> None:
await delete_vfolders(
await delete_vfolders_in_bgtask(
[VFolderDeletionInfo(row.vfid, row.host) for row in alive_vfolders],
storage_manager=graph_ctx.storage_manager,
db=graph_ctx.db,
Expand Down
2 changes: 1 addition & 1 deletion src/ai/backend/manager/models/vfolder.py
Original file line number Diff line number Diff line change
Expand Up @@ -1756,7 +1756,7 @@ class Meta:
items = graphene.List(VirtualFolder, required=True)


async def delete_vfolders(
async def delete_vfolders_in_bgtask(
requested_vfolders: Sequence[VFolderDeletionInfo],
*,
storage_manager: StorageSessionManager,
Expand Down

0 comments on commit 1e4c934

Please sign in to comment.