From 818fa081795cda7b6bad97f35d2bc219f445c0c5 Mon Sep 17 00:00:00 2001
From: Peter Sharpe <pds@mit.edu>
Date: Wed, 17 Jan 2024 15:34:20 -0500
Subject: [PATCH] fix bug in np.gradient with multi-axis gradients

---
 aerosandbox/numpy/calculus.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/aerosandbox/numpy/calculus.py b/aerosandbox/numpy/calculus.py
index 7f3ad086..55345585 100644
--- 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
                 )