Bug Fixed: dopri and rkf78 index error

alvinng4 · Jan 25, 2024 · 9d01c2f · 9d01c2f
1 parent 2815106
commit 9d01c2f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/gravity_sim/simulator.py b/gravity_sim/simulator.py
@@ -341,8 +341,8 @@ def rk_embedded(
     # fac_min = 0.33
     # fac = 0.38**(1.0 / (1.0 + min_power))
 
-    vk = np.zeros((max_power + 1, objects_count, 3))
-    xk = np.zeros((max_power + 1, objects_count, 3))
+    vk = np.zeros((stages, objects_count, 3))
+    xk = np.zeros((stages, objects_count, 3))
     integrate = True
     while integrate:
         for stage in range(stages):