From 185497c56391abf3fe8e2a59cb8f7e57cc718297 Mon Sep 17 00:00:00 2001 From: "James R. Maddison" Date: Fri, 1 Nov 2024 08:59:27 +0000 Subject: [PATCH] Fix --- docs/source/examples/0_getting_started.ipynb | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/docs/source/examples/0_getting_started.ipynb b/docs/source/examples/0_getting_started.ipynb index 4b0aa39..c1d97c6 100644 --- a/docs/source/examples/0_getting_started.ipynb +++ b/docs/source/examples/0_getting_started.ipynb @@ -138,7 +138,7 @@ "outputs": [], "source": [ "def pv_flux(model):\n", - " u = -model.fields[\"psi\"] @ model.grid.D_y.T\n", + " u = -model.grid.D_y(model.fields[\"psi\"]\n", " x = jnp.array((-model.grid.L_x + jnp.cbrt(model.nu / model.beta),))\n", " y = jnp.linspace(0, model.grid.L_y, 1025)\n", " u_munk = model.grid.interpolate(u, x, y)[0, :]\n",