cover other type's fields

lablup · Jan 15, 2025 · df413fc · df413fc
1 parent a6ef9d3
commit df413fc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/ai/backend/manager/models/session.py b/src/ai/backend/manager/models/session.py
@@ -1678,7 +1678,7 @@ def parse_row(cls, ctx: GraphQueryContext, row: Row) -> Mapping[str, Any]:
             "scaling_group": row.scaling_group_name,
             "service_ports": row.main_kernel.service_ports,
             "mounts": [mount.name for mount in row.vfolder_mounts],
-            "vfolder_mounts": row.vfolder_mounts,
+            "vfolder_mounts": [vf.vfid.folder_id for vf in row.vfolder_mounts],
             "occupying_slots": row.occupying_slots.to_json(),
             "occupied_slots": row.occupying_slots.to_json(),
             "requested_slots": row.requested_slots.to_json(),