Nicen up the animation

Author: danshapero

notebooks/rainier-monolithic.ipynb

@@ -7,7 +7,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "from tqdm.notebook import trange\n",
+    "from tqdm.notebook import trange, tqdm\n",
     "import numpy as np\n",
     "from numpy import pi as π\n",
     "import matplotlib.pyplot as plt\n",
@@ -700,20 +700,28 @@
    "source": [
     "%%capture\n",
     "\n",
-    "from matplotlib.animation import FuncAnimation\n",
-    "\n",
     "fig, axes = plt.subplots()\n",
     "axes.set_aspect(\"equal\")\n",
     "axes.set_xlim((0, 10e3))\n",
     "axes.set_ylim((0, 10e3))\n",
     "colors = firedrake.tripcolor(hs[0], vmax=130.0, num_sample_points=4, axes=axes)\n",
-    "fig.colorbar(colors)\n",
+    "fig.colorbar(colors);"
+   ]
+  },
+  {
+   "cell_type": "code",
+   "execution_count": null,
+   "id": "66704ac7-81e5-4586-be98-86f821228ca4",
+   "metadata": {},
+   "outputs": [],
+   "source": [
+    "from matplotlib.animation import FuncAnimation\n",
     "\n",
     "fn_plotter = firedrake.FunctionPlotter(mesh, num_sample_points=4)\n",
     "def animate(h):\n",
     "    colors.set_array(fn_plotter(h))\n",
     "\n",
-    "animation = FuncAnimation(fig, animate, hs, interval=1e3/60)"
+    "animation = FuncAnimation(fig, animate, tqdm(hs), interval=1e3/60)"
    ]
   },
   {