Skip to content

Commit 0eb7bb7

Browse files
authored
Merge pull request #91 from QuanEstimation/fix-ext-decay
fix python extension decay type convertion
2 parents ff2a780 + 5a7e7c0 commit 0eb7bb7

File tree

1 file changed

+11
-11
lines changed

1 file changed

+11
-11
lines changed

ext/QuanEstimationPyExt.jl

+11-11
Original file line numberDiff line numberDiff line change
@@ -21,8 +21,8 @@ QuanEstimation.Lindblad(
2121
pyconvert(Vector{Vector{Float64}}, ctrl),
2222
pyconvert(Matrix{ComplexF64}, ρ0),
2323
pyconvert(Vector, tspan),
24-
pyconvert(Vector{Matrix}, decay_opt),
25-
pyconvert(Vector, γ);
24+
pyconvert(Vector{Matrix{ComplexF64}}, decay_opt),
25+
pyconvert(Vector{Float64}, γ);
2626
kwargs...,
2727
)
2828

@@ -43,8 +43,8 @@ QuanEstimation.Lindblad(
4343
pyconvert(Vector{Vector{Float64}}, ctrl),
4444
pyconvert(Vector{ComplexF64}, ψ0),
4545
pyconvert(Vector, tspan),
46-
pyconvert(Vector{Matrix}, decay_opt),
47-
pyconvert(Vector, γ);
46+
pyconvert(Vector{Matrix{ComplexF64}}, decay_opt),
47+
pyconvert(Vector{Float64}, γ);
4848
kwargs...,
4949
)
5050

@@ -61,8 +61,8 @@ QuanEstimation.Lindblad(
6161
pyconvert(Vector{Matrix}, dH),
6262
pyconvert(Vector{ComplexF64}, ψ0),
6363
pyconvert(Vector, tspan),
64-
pyconvert(Vector{Matrix}, decay_opt),
65-
pyconvert(Vector, γ);
64+
pyconvert(Vector{Matrix{ComplexF64}}, decay_opt),
65+
pyconvert(Vector{Float64}, γ);
6666
kwargs...,
6767
)
6868

@@ -79,8 +79,8 @@ QuanEstimation.Lindblad(
7979
pyconvert(Vector{Matrix}, dH),
8080
pyconvert(Matrix{ComplexF64}, ρ0),
8181
pyconvert(Vector, tspan),
82-
pyconvert(Vector{Matrix}, decay_opt),
83-
pyconvert(Vector, γ);
82+
pyconvert(Vector{Matrix{ComplexF64}}, decay_opt),
83+
pyconvert(Vector{Float64}, γ);
8484
kwargs...,
8585
)
8686

@@ -94,7 +94,7 @@ QuanEstimation.Lindblad(
9494
pyconvert(Matrix{ComplexF64}, H0),
9595
pyconvert(Vector{Matrix}, dH),
9696
pyconvert(Vector{ComplexF64}, ψ0),
97-
pyconvert(Vector, tspan);
97+
pyconvert(Vector{Float64}, tspan);
9898
kwargs...,
9999
)
100100

@@ -108,7 +108,7 @@ QuanEstimation.Lindblad(
108108
pyconvert(Matrix{ComplexF64}, H0),
109109
pyconvert(Vector{Matrix}, dH),
110110
pyconvert(Matrix{ComplexF64}, ρ0),
111-
pyconvert(Vector, tspan);
111+
pyconvert(Vector{Float64}, tspan);
112112
kwargs...,
113113
)
114114

@@ -151,7 +151,7 @@ QuanEstimation.expm_py(
151151
pyconvert(Matrix{ComplexF64}, H0),
152152
pyconvert(Vector{Matrix{ComplexF64}}, dH),
153153
pyconvert(Vector{Matrix{ComplexF64}}, decay_opt),
154-
pyconvert(Vector, γ),
154+
pyconvert(Vector{Float64}, γ),
155155
pyconvert(Vector{Matrix{ComplexF64}}, Hc),
156156
pyconvert(Vector{Vector{Float64}}, ctrl),
157157
)

0 commit comments

Comments
 (0)