Skip to content

Commit 77bba70

Browse files
committed
Better type detection in complex
1 parent 2586ad7 commit 77bba70

File tree

1 file changed

+6
-4
lines changed

1 file changed

+6
-4
lines changed

src/states/finitemps.jl

Lines changed: 6 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -325,10 +325,12 @@ function Base.complex(mps::FiniteMPS)
325325
ARs = _complex_if_not_missing.(mps.ARs)
326326
Cs = _complex_if_not_missing.(mps.Cs)
327327
ACs = _complex_if_not_missing.(mps.ACs)
328-
return FiniteMPS(collect(Union{Missing,eltype(ALs)}, ALs),
329-
collect(Union{Missing,eltype(ARs)}, ARs),
330-
collect(Union{Missing,eltype(ACs)}, ACs),
331-
collect(Union{Missing,eltype(Cs)}, Cs))
328+
TA = Base.promote_op(complex, site_type(mps))
329+
TB = Base.promote_op(complex, bond_type(mps))
330+
return FiniteMPS(collect(Union{Missing,TA}, ALs),
331+
collect(Union{Missing,TA}, ARs),
332+
collect(Union{Missing,TA}, ACs),
333+
collect(Union{Missing,TB}, Cs))
332334
end
333335
334336
@inline function Base.getindex(ψ::FiniteMPS, I::AbstractUnitRange)

0 commit comments

Comments
 (0)