fix(Core/typetransforms): unwrapping Type in Union

tecosaur · Oct 13, 2024 · 138ccd1 · 138ccd1
1 parent fb1bd40
commit 138ccd1
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/Core/src/interaction/typetransforms.jl b/Core/src/interaction/typetransforms.jl
@@ -163,7 +163,11 @@ function typevariants(T::Type)::Vector{Type}
     elseif T isa UnionAll || (T isa DataType && T.name.name != :Type)
         [T]
     elseif T isa Union
-        Base.uniontypes(T)
+        Ta = Type[]
+        for Tu in Base.uniontypes(T)
+            append!(Ta, typevariants(Tu))
+        end
+        Ta
     elseif T isa Type{<:Any}
         typevariants(first(T.parameters))
     else
@@ -212,7 +216,7 @@ function transformersigs(L::Type{<:DataLoader}, desired::Type)
         if Tloader isa TypeVar
             Tloader = Tloader.ub
         end
-        if Tout1 == Type{Tin}
+        if Tout1 == Type{Tin} && Tin isa TypeVar || Tout1 == Type
             push!(types, (Tloader, desired, desired))
         else
             for Tout in typevariants(Tout1)