infer default for union shapes, now support more union types

widmogrod · Oct 27, 2023 · 7a2b62d · 7a2b62d
1 parent c7f83f6
commit 7a2b62d
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/infer_defaults.go b/infer_defaults.go
@@ -116,7 +116,7 @@ func (f *InferredInfo) StructShapeWith(name string) *shape.StructLike {
 func (f *InferredInfo) RetrieveUnions() []*shape.UnionLike {
 	result := make([]*shape.UnionLike, 0)
 	for unionName, possibleVariants := range f.possibleVariantTypes {
-		var variants []*shape.StructLike
+		var variants []shape.Shape
 		for _, variant := range possibleVariants {
 			variants = append(variants, f.shapes[variant])
 		}