fix bug in np.gradient with multi-axis gradients

peterdsharpe · Jan 17, 2024 · 818fa08 · 818fa08
1 parent f6fa352
commit 818fa08
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/aerosandbox/numpy/calculus.py b/aerosandbox/numpy/calculus.py
@@ -93,7 +93,7 @@ def gradient(
             return [
                 gradient(
                     f,
-                    *varargs,
+                    varargs[axis_i],
                     axis=axis_i,
                     edge_order=edge_order
                 )