fix: Broken ComputeContainer GQL (#3042) (#3071)

Co-authored-by: Gyubong Lee <[email protected]> Co-authored-by: Joongi Kim <[email protected]>
lablup · Nov 12, 2024 · c502915 · c502915
1 parent 213348a
commit c502915
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 3 deletions.
diff --git a/changes/3042.fix.md b/changes/3042.fix.md
@@ -0,0 +1 @@
+Fix regression of `ComputeContainer` GraphQL queries due to newly introduced relationship fields
diff --git a/src/ai/backend/manager/models/kernel.py b/src/ai/backend/manager/models/kernel.py
@@ -1024,9 +1024,14 @@ async def batch_load_detail(
             .where(
                 (KernelRow.id.in_(container_ids)),
             )
-            .options(selectinload(KernelRow.group_row))
-            .options(selectinload(KernelRow.user_row))
-            .options(selectinload(KernelRow.image_row))
+            .options(
+                noload("*"),
+                selectinload(
+                    KernelRow.group_row,
+                    KernelRow.user_row,
+                    KernelRow.image_row,
+                ),
+            )
         )
         if domain_name is not None:
             query = query.where(KernelRow.domain_name == domain_name)