Fix

jrmaddison · Nov 1, 2024 · 185497c · 185497c
1 parent 736a4b4
commit 185497c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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",