Fix example notebooks: remove broken state access and steady-state plots

Author: milanofthe

docs/source/examples/flash_distillation.ipynb

@@ -154,33 +154,7 @@
    "execution_count": null,
    "metadata": {},
    "outputs": [],
-   "source": [
-    "time, signals = scp.read()\n",
-    "\n",
-    "fig, (ax1, ax2) = plt.subplots(1, 2, figsize=(12, 5))\n",
-    "\n",
-    "# Dynamic tray composition evolution\n",
-    "for i in range(N_trays):\n",
-    "    ax1.plot(time, signals[i], label=f\"Tray {i+1}\")\n",
-    "ax1.set_xlabel(\"Time [s]\")\n",
-    "ax1.set_ylabel(r\"$x$ (light component)\")\n",
-    "ax1.set_title(\"Tray Compositions Over Time\")\n",
-    "ax1.legend()\n",
-    "ax1.grid(True, alpha=0.3)\n",
-    "\n",
-    "# Steady-state composition profile\n",
-    "x_profile = [tray.engine.state[0] for tray in trays]\n",
-    "tray_nums = list(range(1, N_trays + 1))\n",
-    "\n",
-    "ax2.plot(tray_nums, x_profile, \"o-\", markersize=8)\n",
-    "ax2.set_xlabel(\"Tray number (top to bottom)\")\n",
-    "ax2.set_ylabel(r\"$x$ (light component)\")\n",
-    "ax2.set_title(\"Composition Profile (Steady State)\")\n",
-    "ax2.grid(True, alpha=0.3)\n",
-    "\n",
-    "plt.tight_layout()\n",
-    "plt.show()"
-   ]
+   "source": "time, signals = scp.read()\n\nfig, (ax1, ax2) = plt.subplots(1, 2, figsize=(12, 5))\n\n# Dynamic tray composition evolution\nfor i in range(N_trays):\n    ax1.plot(time, signals[i], label=f\"Tray {i+1}\")\nax1.set_xlabel(\"Time [s]\")\nax1.set_ylabel(r\"$x$ (light component)\")\nax1.set_title(\"Tray Compositions Over Time\")\nax1.legend()\nax1.grid(True, alpha=0.3)\n\n# Steady-state composition profile\nx_profile = [tray.state[0] for tray in trays]\ntray_nums = list(range(1, N_trays + 1))\n\nax2.plot(tray_nums, x_profile, \"o-\", markersize=8)\nax2.set_xlabel(\"Tray number (top to bottom)\")\nax2.set_ylabel(r\"$x$ (light component)\")\nax2.set_title(\"Composition Profile (Steady State)\")\nax2.grid(True, alpha=0.3)\n\nplt.tight_layout()\nplt.show()"
   },
   {
    "cell_type": "markdown",

docs/source/examples/heat_exchanger.ipynb

@@ -68,39 +68,7 @@
    "execution_count": null,
    "metadata": {},
    "outputs": [],
-   "source": [
-    "time, signals = scp.read()\n",
-    "\n",
-    "fig, (ax1, ax2) = plt.subplots(1, 2, figsize=(12, 5))\n",
-    "\n",
-    "# Outlet temperatures over time\n",
-    "ax1.plot(time, signals[0], label=\"Hot outlet\")\n",
-    "ax1.plot(time, signals[1], label=\"Cold outlet\")\n",
-    "ax1.axhline(370, color=\"gray\", ls=\"--\", alpha=0.4, label=\"Hot inlet\")\n",
-    "ax1.axhline(290, color=\"gray\", ls=\"-.\", alpha=0.4, label=\"Cold inlet\")\n",
-    "ax1.set_xlabel(\"Time [s]\")\n",
-    "ax1.set_ylabel(\"Temperature [K]\")\n",
-    "ax1.set_title(\"Outlet Temperatures\")\n",
-    "ax1.legend()\n",
-    "ax1.grid(True, alpha=0.3)\n",
-    "\n",
-    "# Steady-state temperature profile along the exchanger\n",
-    "state = hx.engine.x0\n",
-    "T_h_profile = state[0::2]\n",
-    "T_c_profile = state[1::2]\n",
-    "cells = np.arange(1, len(T_h_profile) + 1)\n",
-    "\n",
-    "ax2.plot(cells, T_h_profile, \"o-\", label=\"Hot stream\")\n",
-    "ax2.plot(cells, T_c_profile, \"s-\", label=\"Cold stream\")\n",
-    "ax2.set_xlabel(\"Cell number\")\n",
-    "ax2.set_ylabel(\"Temperature [K]\")\n",
-    "ax2.set_title(\"Temperature Profile (Steady State)\")\n",
-    "ax2.legend()\n",
-    "ax2.grid(True, alpha=0.3)\n",
-    "\n",
-    "plt.tight_layout()\n",
-    "plt.show()"
-   ]
+   "source": "time, signals = scp.read()\n\nfig, ax = plt.subplots(figsize=(8, 5))\n\nax.plot(time, signals[0], label=\"Hot outlet\")\nax.plot(time, signals[1], label=\"Cold outlet\")\nax.axhline(370, color=\"gray\", ls=\"--\", alpha=0.4, label=\"Hot inlet\")\nax.axhline(290, color=\"gray\", ls=\"-.\", alpha=0.4, label=\"Cold inlet\")\nax.set_xlabel(\"Time [s]\")\nax.set_ylabel(\"Temperature [K]\")\nax.set_title(\"Heat Exchanger Outlet Temperatures\")\nax.legend()\nax.grid(True, alpha=0.3)\n\nplt.tight_layout()\nplt.show()"
   },
   {
    "cell_type": "markdown",