Skip to content

Commit a222ddf

Browse files
authored
fix(BA-1167): Wrong custom image owner check logic (#4181)
1 parent 31380e2 commit a222ddf

File tree

2 files changed

+3
-1
lines changed

2 files changed

+3
-1
lines changed

changes/4181.fix.md

+1
Original file line numberDiff line numberDiff line change
@@ -0,0 +1 @@
1+
Fix wrong custom image owner check logic.

src/ai/backend/manager/models/image.py

+2-1
Original file line numberDiff line numberDiff line change
@@ -798,7 +798,8 @@ def set_resource_limit(
798798
self._resources = resources
799799

800800
def is_owned_by(self, user_id: UUID) -> bool:
801-
return self.customized and self.labels["ai.backend.customized-image.owner"] == str(user_id)
801+
owner_label = self.labels["ai.backend.customized-image.owner"]
802+
return self.customized and owner_label.split(":")[1] == str(user_id)
802803

803804
def to_dataclass(self) -> ImageData:
804805
from ai.backend.manager.data.image.types import ImageData

0 commit comments

Comments
 (0)