Fix shape descriptor for polymorphic shape elements (#7949)

The `ShapeElement.source_type` field is now properly populated with origin object type.
geldata · Oct 31, 2024 · f7e4758 · f7e4758
1 parent b9ba3b6
commit f7e4758
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/edb/server/compiler/sertypes.py b/edb/server/compiler/sertypes.py
@@ -551,7 +551,8 @@ def _describe_object_shape(
         link_props.append(False)
         links.append(not ptr.is_property(ctx.schema))
         cardinalities.append(cardinality_from_ptr(ptr, ctx.schema))
-        ptr_source = ptr.get_source(ctx.schema)
+        ctx.schema, material_ptr = ptr.material_type(ctx.schema)
+        ptr_source = material_ptr.get_source(ctx.schema)
         assert isinstance(ptr_source, s_objtypes.ObjectType)
         ctx.schema, ptr_source = ptr_source.material_type(ctx.schema)
         assert ptr_source is not None