Skip to content

Commit 1d5ea7c

Browse files
MinJuTurclaude
andcommitted
refactor(BA-3687): use UUID instead of uuid.UUID in image types
Follow internal convention of using UUID directly. Co-Authored-By: Claude Opus 4.5 <noreply@anthropic.com>
1 parent d4889f5 commit 1d5ea7c

File tree

1 file changed

+5
-5
lines changed
  • src/ai/backend/manager/data/image

1 file changed

+5
-5
lines changed

src/ai/backend/manager/data/image/types.py

Lines changed: 5 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -1,9 +1,9 @@
11
import enum
2-
import uuid
32
from dataclasses import dataclass, field
43
from datetime import datetime
54
from decimal import Decimal
65
from typing import TYPE_CHECKING, NamedTuple, Optional
6+
from uuid import UUID
77

88
from ai.backend.common.types import CIStrEnum, ImageCanonical, ImageID
99

@@ -50,7 +50,7 @@ class ImageData:
5050
created_at: Optional[datetime] = field(compare=False)
5151
tag: Optional[str]
5252
registry: str
53-
registry_id: uuid.UUID
53+
registry_id: UUID
5454
architecture: str
5555
config_digest: str
5656
size_bytes: int
@@ -60,7 +60,7 @@ class ImageData:
6060
labels: ImageLabelsData
6161
resources: ImageResourcesData
6262
status: ImageStatus
63-
owner_id: Optional[uuid.UUID] = field(default=None)
63+
owner_id: Optional[UUID] = field(default=None)
6464

6565

6666
@dataclass
@@ -88,7 +88,7 @@ class ImageDataWithDetails:
8888
tags: list[KVPair]
8989
version: Optional[str]
9090
registry: str
91-
registry_id: uuid.UUID
91+
registry_id: UUID
9292
type: ImageType
9393
architecture: str
9494
is_local: bool
@@ -131,5 +131,5 @@ class RescanImagesResult:
131131

132132
@dataclass
133133
class ImageAliasData:
134-
id: uuid.UUID = field(compare=False)
134+
id: UUID = field(compare=False)
135135
alias: str

0 commit comments

Comments
 (0)