Cosmetic change

Kolaru · Feb 14, 2024 · d56421b · d56421b
1 parent 5c1d591
commit d56421b
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/NormalModes.jl b/src/NormalModes.jl
@@ -162,12 +162,12 @@ momentum.
 """
 function StatsBase.sample(rng::AbstractRNG, nm::NormalDecomposition, n_samples)
     hbar = 1  # Atomic units
+    X = nm.M * nm.U
     Δx_dist = MvNormal(Diagonal(1/2 * (hbar ./ nm.ωs)))
     Δp_dist = MvNormal(Diagonal(1/2 * hbar * nm.ωs))
-    MU = nm.M * nm.U
 
-    Δx = MU * rand(rng, Δx_dist, n_samples)
-    Δp = inv(nm.M)^2 * MU * rand(rng, Δp_dist, n_samples)
+    Δx = X * rand(rng, Δx_dist, n_samples)
+    Δp = inv(nm.M)^2 * X * rand(rng, Δp_dist, n_samples)
 
     return Δx * aunit(u"m"), Δp * aunit(u"kg*m/s")
 end