fix: Improve _in_project_scope using relationship

lablup · Nov 20, 2024 · 28107a1 · 28107a1
1 parent 7117c6b
commit 28107a1
Showing 1 changed file with 2 additions and 9 deletions.
diff --git a/src/ai/backend/manager/models/image.py b/src/ai/backend/manager/models/image.py
@@ -40,9 +40,6 @@
 
 from ..api.exceptions import ImageNotFound
 from ..container_registry import get_container_registry_cls
-from ..models.association_container_registries_groups import (
-    AssociationContainerRegistriesGroupsRow,
-)
 from ..models.container_registry import ContainerRegistryRow
 from .base import (
     GUID,
@@ -796,12 +793,8 @@ async def _in_project_scope(
                     ContainerRegistryRow.is_global == true(),
                     sa.and_(
                         ContainerRegistryRow.is_global == false(),
-                        sa.exists().where(
-                            (AssociationContainerRegistriesGroupsRow.group_id == scope.project_id)
-                            & (
-                                AssociationContainerRegistriesGroupsRow.registry_id
-                                == ImageRow.registry_id
-                            )
+                        ContainerRegistryRow.association_container_registries_groups_rows.any(
+                            group_id=scope.project_id
                         ),
                     ),
                 )