fix median/nanmedian swap

adrn · Jun 21, 2021 · eb4aea4 · eb4aea4
1 parent e0f8e20
commit eb4aea4
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/gala/dynamics/actionangle_staeckel.py b/gala/dynamics/actionangle_staeckel.py
@@ -52,6 +52,6 @@ def get_staeckel_fudge_delta(potential, w, median=True):
 
     # Median over time if the inputs were orbits
     if len(delta.shape) > 1 and median:
-        delta = np.median(delta, axis=1)
+        delta = np.nanmedian(delta, axis=1)
 
     return delta * potential.units['length']