From fb2a42233332dede11a972cbff2a09a34d40b2f9 Mon Sep 17 00:00:00 2001 From: Nathaniel Starkman Date: Thu, 18 Jan 2024 16:57:20 -0500 Subject: [PATCH] Pass the correct quantities to run_vmap (#62) Signed-off-by: nstarman --- src/galax/dynamics/mockstream/_mockstream_generator.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/galax/dynamics/mockstream/_mockstream_generator.py b/src/galax/dynamics/mockstream/_mockstream_generator.py index 3ac6a4fe..a1cf5afc 100644 --- a/src/galax/dynamics/mockstream/_mockstream_generator.py +++ b/src/galax/dynamics/mockstream/_mockstream_generator.py @@ -141,7 +141,7 @@ def run( ) if vmapped: - lead_arm_qp, trail_arm_qp = self._run_vmap(ts, prog_w0, prog_mass) + lead_arm_qp, trail_arm_qp = self._run_vmap(ts, mock0_lead, mock0_trail) else: lead_arm_qp, trail_arm_qp = self._run_scan(ts, mock0_lead, mock0_trail)