fix bugs for multi-dimension trajectories

EigenSolver · Aug 20, 2024 · 77cef6d · 77cef6d
1 parent fc0af56
commit 77cef6d
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/SpinShuttling.jl b/src/SpinShuttling.jl
@@ -76,7 +76,8 @@ function OneSpinModel(Ψ::Vector{<:Complex}, T::Real, N::Int,
     B::GaussianRandomField, x::Function)
 
     t = range(0, T, N)
-    P = collect(zip(t, x.(t)))
+    f(x::Function, t::Real) = (t, x(t)...)
+    P = f.(x, t)
     R = RandomFunction(P, B)
     model = ShuttlingModel(1, Ψ, T, N, B, [x], R)
     return model