fix sparse conversion

qutip · Oct 11, 2024 · edeef97 · edeef97
1 parent 3505eb0
commit edeef97
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/evolution.jl b/src/evolution.jl
@@ -254,7 +254,7 @@ function HEOMsolve(
         flush(stdout)
     end
     sol = solve(prob, solver)
-    ADOs_list = map(ρvec -> ADOs(Vector(ρvec), M.dims, M.N, M.parity), sol.u)
+    ADOs_list = map(ρvec -> ADOs(sparse(Vector(ρvec)), M.dims, M.N, M.parity), sol.u)
 
     # save ADOs to file
     if filename != ""

diff --git a/src/steadystate.jl b/src/steadystate.jl
@@ -44,7 +44,7 @@ function steadystate(
         flush(stdout)
     end
 
-    return ADOs(Vector(sol.u), M.dims, M.N, M.parity)
+    return ADOs(sparse(Vector(sol.u)), M.dims, M.N, M.parity)
 end
 
 @doc raw"""