From b7675e6e1ec839ebae7c6080fd6afa80d1b12cd5 Mon Sep 17 00:00:00 2001 From: YigitElma Date: Sat, 19 Oct 2024 21:10:42 -0400 Subject: [PATCH] fix zeta phi problem causing nans --- tests/test_equilibrium.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/tests/test_equilibrium.py b/tests/test_equilibrium.py index cfab9107d..51202e3f8 100644 --- a/tests/test_equilibrium.py +++ b/tests/test_equilibrium.py @@ -103,7 +103,7 @@ def test_map_coordinates_derivative(): def foo(params, in_coords): out = eq.map_coordinates( in_coords, - ("rho", "alpha", "zeta"), # for this test, zeta==phi + ("rho", "alpha", "phi"), ("rho", "theta_PEST", "zeta"), np.array([rho, theta, zeta]).T, params, @@ -123,9 +123,9 @@ def foo(params, in_coords): # this time _map_clebsch_coordinates is called inside map_coordinates @jax.jit def foo(params, in_coords): - out, _ = eq.map_coordinates( + out, (_, _) = eq.map_coordinates( in_coords, - ("rho", "alpha", "zeta"), # for this test, zeta==phi + ("rho", "alpha", "phi"), ("rho", "theta", "zeta"), np.array([rho, theta, zeta]).T, params,