added to_dense call to sesolve_map to ensure all initial states are dense.

Author: Gavin-Rockwood

src/time_evolution/sesolve.jl

@@ -235,6 +235,9 @@ function sesolve_map(
     kwargs...,
 ) where {X<:Union{Ket,Operator}}
     # mapping initial states and parameters
+
+    ψ0 = map(to_dense, ψ0) # Convert all initial states to dense vectors
+
     ψ0_iter = map(get_data, ψ0)
     if params isa NullParameters
         iter = collect(Iterators.product(ψ0_iter, [params])) |> vec # convert nx1 Matrix into Vector

testing.ipynb

@@ -517,15 +517,15 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 7,
+   "execution_count": 3,
    "id": "d90f8989",
    "metadata": {},
    "outputs": [
     {
      "name": "stderr",
      "output_type": "stream",
      "text": [
-      "\u001b[32m[sesolve] 100%|███████████████████████████| Time: 0:00:00 ( 0.51 μs/it)\u001b[39m\u001b[K\n"
+      "\u001b[32m[sesolve] 100%|███████████████████████████| Time: 0:00:00 (64.34 μs/it)\u001b[39m\u001b[K\n"
      ]
     },
     {
@@ -554,15 +554,15 @@
   },
   {
    "cell_type": "code",
-   "execution_count": 17,
+   "execution_count": 5,
    "id": "126d75af",
    "metadata": {},
    "outputs": [
     {
      "name": "stderr",
      "output_type": "stream",
      "text": [
-      "\u001b[32m[sesolve_map] 100%|███████████████████████| Time: 0:00:00 ( 0.11 ms/it)\u001b[39m\u001b[K\n"
+      "\u001b[32m[sesolve_map] 100%|███████████████████████| Time: 0:00:00 ( 0.66 ms/it)\u001b[39m\u001b[K\n"
      ]
     },
     {
@@ -594,7 +594,7 @@
    ],
    "source": [
     "H = sigmax()\n",
-    "ψ0 = to_dense(qeye(2))\n",
+    "ψ0 = (qeye(2))\n",
     "#ψ0 = fock(2,1)\n",
     "tlist = 0:0.1:10.0\n",
     "sol = sesolve_map(H, [ψ0,ψ0], tlist)"