Typo in trojan calculation (#127)

Caltech-IPAC · Sep 27, 2024 · fada65b · fada65b
1 parent ba639cc
commit fada65b
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/src/examples/earth_trojan.py b/src/examples/earth_trojan.py
@@ -60,7 +60,10 @@
 
     energy_vals.append(total / m_steps)
 
-cur_energy = 3 * (elem_a.semi_major - 1) ** 2 / (8.0 * mass_planet) + energy_vals[0]
+cur_energy = (
+    3 * (elem_a.semi_major - elem_b.semi_major) ** 2 / (8.0 * mass_planet)
+    + energy_vals[0]
+)
 
 # energy vals start from the current positions and go a full orbit.
 # These may be unrolled so that the Earth is in the middle, the following