chore: change name of dto

lablup · Feb 14, 2025 · 5cdc8c0 · 5cdc8c0
1 parent 749d43e
commit 5cdc8c0
Show file tree

Hide file tree

Showing 3 changed files with 7 additions and 7 deletions.
diff --git a/src/ai/backend/common/dto/manager/field.py b/src/ai/backend/common/dto/manager/field.py
@@ -6,7 +6,7 @@
 from ai.backend.common.types import VFolderUsageMode
 
 
-class VFolderPermissionField(enum.StrEnum):
+class VFolderMountPermissionField(enum.StrEnum):
     READ_ONLY = "ro"
     READ_WRITE = "rw"
     RW_DELETE = "wd"
@@ -38,7 +38,7 @@ class VFolderItemField(BaseModel):
     host: str
     usage_mode: VFolderUsageMode
     created_at: str
-    permission: VFolderPermissionField
+    permission: VFolderMountPermissionField
     max_size: int
     creator: str
     ownership_type: VFolderOwnershipTypeField

diff --git a/src/ai/backend/common/dto/manager/request.py b/src/ai/backend/common/dto/manager/request.py
@@ -4,7 +4,7 @@
 from pydantic import AliasChoices, BaseModel, Field
 
 from ai.backend.common import typed_validators as tv
-from ai.backend.common.dto.manager.field import VFolderPermissionField
+from ai.backend.common.dto.manager.field import VFolderMountPermissionField
 from ai.backend.common.types import VFolderUsageMode
 
 
@@ -17,7 +17,7 @@ class VFolderCreateReq(BaseModel):
         default=None,
     )
     usage_mode: VFolderUsageMode = Field(default=VFolderUsageMode.GENERAL)
-    permission: VFolderPermissionField = Field(default=VFolderPermissionField.READ_WRITE)
+    permission: VFolderMountPermissionField = Field(default=VFolderMountPermissionField.READ_WRITE)
     unmanaged_path: Optional[str] = Field(
         validation_alias=AliasChoices("unmanaged_path", "unmanagedPath"),
         default=None,

diff --git a/src/ai/backend/manager/models/vfolder.py b/src/ai/backend/manager/models/vfolder.py
@@ -41,9 +41,9 @@
 from ai.backend.common.bgtask import ProgressReporter
 from ai.backend.common.defs import MODEL_VFOLDER_LENGTH_LIMIT
 from ai.backend.common.dto.manager.field import (
+    VFolderMountPermissionField,
     VFolderOperationStatusField,
     VFolderOwnershipTypeField,
-    VFolderPermissionField,
 )
 from ai.backend.common.types import (
     MountPermission,
@@ -177,8 +177,8 @@ class VFolderPermission(enum.StrEnum):
     RW_DELETE = "wd"
     OWNER_PERM = "wd"  # resolved as RW_DELETE
 
-    def to_field(self) -> VFolderPermissionField:
-        return VFolderPermissionField(self)
+    def to_field(self) -> VFolderMountPermissionField:
+        return VFolderMountPermissionField(self)
 
 
 class VFolderPermissionValidator(t.Trafaret):