Fix _pode_samples.

JuliaGNI · Nov 21, 2024 · b4a8509 · b4a8509
1 parent 07cbeac
commit b4a8509
Showing 1 changed file with 5 additions and 2 deletions.
diff --git a/src/harmonic_oscillator.jl b/src/harmonic_oscillator.jl
@@ -109,9 +109,12 @@ module HarmonicOscillator
 
         qsamples = vec(collect.(collect(Base.Iterators.product(qs...))))
         psamples = vec(collect.(collect(Base.Iterators.product(ps...))))
-        samples = vec(collect(Base.Iterators.product(qsamples, psamples)))
+        zsamples = Base.Iterators.product(qsamples, psamples)
 
-        (q = qsamples, p = psamples)
+        return (
+            q = vec([zs[1] for zs in zsamples]),
+            p = vec([zs[2] for zs in zsamples]),
+        )
     end