use grid.add_field

angelmons · Oct 20, 2024 · d6aaddf · d6aaddf
1 parent 40356c7
commit d6aaddf
Showing 1 changed file with 3 additions and 6 deletions.
diff --git a/docs/source/tutorials/river_flow_dynamics/river_flow_dynamics_tutorial.ipynb b/docs/source/tutorials/river_flow_dynamics/river_flow_dynamics_tutorial.ipynb
@@ -155,8 +155,7 @@
    "outputs": [],
    "source": [
     "# The grid represents a basic rectangular channel with slope equal to 0.01 m/m\n",
-    "te = grid.add_zeros(\"topographic__elevation\", at=\"node\")\n",
-    "te += 1.0 - channel_slope * grid.x_of_node\n",
+    "te = grid.add_field(\"topographic__elevation\", 1.0 - channel_slope * grid.x_of_node, at=\"node\")\n",
     "te[grid.y_of_node > 1.5] = 2.5\n",
     "te[grid.y_of_node < 0.5] = 2.5"
    ]
@@ -198,8 +197,7 @@
     "vel = grid.add_zeros(\"surface_water__velocity\", at=\"link\")\n",
     "\n",
     "# Calculating the initial water surface elevation from water depth and topographic elevation\n",
-    "wse = grid.add_zeros(\"surface_water__elevation\", at=\"node\")\n",
-    "wse += h + te"
+    "wse = grid.add_field(\"surface_water__elevation\", te, at=\"node\")"
    ]
   },
   {
@@ -428,8 +426,7 @@
     "vel = grid.add_zeros(\"surface_water__velocity\", at=\"link\")\n",
     "\n",
     "# Calculating the initial water surface elevation from water depth and topographic elevation\n",
-    "wse = grid.add_zeros(\"surface_water__elevation\", at=\"node\")\n",
-    "wse += h + te"
+    "wse = grid.add_field(\"surface_water__elevation\", te, at=\"node\")"
    ]
   },
   {