Update integration_with_ro.ipynb

watertap-org · Sep 12, 2024 · 6c48895 · 6c48895
1 parent bac1d57
commit 6c48895
Showing 1 changed file with 12 additions and 7 deletions.
diff --git a/src/reaktoro_pse/tutorials/integration_with_ro.ipynb b/src/reaktoro_pse/tutorials/integration_with_ro.ipynb
@@ -78,6 +78,7 @@
     "\n",
     "# WaterTAP core components\n",
     "from watertap.property_models.NaCl_prop_pack import NaClParameterBlock\n",
+    "\n",
     "# WaterTAP core components\n",
     "from watertap.property_models.seawater_prop_pack import SeawaterParameterBlock\n",
     "\n",
@@ -133,14 +134,14 @@
     "# create IDAES flowsheet\n",
     "m.fs = FlowsheetBlock(dynamic=False)\n",
     "# can use NaCl or Sea water prop pack\n",
-    "prop_pack='Seawater'\n",
-    "if prop_pack=='Seawater':\n",
+    "prop_pack = \"Seawater\"\n",
+    "if prop_pack == \"Seawater\":\n",
     "    # create NaCl property model\n",
-    "    m.fs.properties=SeawaterParameterBlock()\n",
-    "    bulk_ion='TDS'\n",
+    "    m.fs.properties = SeawaterParameterBlock()\n",
+    "    bulk_ion = \"TDS\"\n",
     "else:\n",
     "    m.fs.properties = NaClParameterBlock()\n",
-    "    bulk_ion='NaCl'\n",
+    "    bulk_ion = \"NaCl\"\n",
     "# build feed\n",
     "m.fs.feed = Feed(property_package=m.fs.properties)"
    ]
@@ -1617,7 +1618,9 @@
     "    )\n",
     "\n",
     "ax.plot(indexes, reaktoro_osm, color=\"red\", label=\"Reaktoro\")\n",
-    "ax.plot(indexes, sewater_prop_pack_osm, color=\"black\", label=f\"{prop_pack} property pack\")\n",
+    "ax.plot(\n",
+    "    indexes, sewater_prop_pack_osm, color=\"black\", label=f\"{prop_pack} property pack\"\n",
+    ")\n",
     "ax.set(xlabel=\"Node position\", ylabel=\"Osmotic pressure (Bar)\")\n",
     "plt.legend()\n",
     "plt.show()"
@@ -1788,7 +1791,9 @@
     "    )\n",
     "\n",
     "ax.plot(indexes, reaktoro_osm, color=\"red\", label=\"Reaktoro\")\n",
-    "ax.plot(indexes, sewater_prop_pack_osm, color=\"black\", label=f\"{prop_pack} property pack\")\n",
+    "ax.plot(\n",
+    "    indexes, sewater_prop_pack_osm, color=\"black\", label=f\"{prop_pack} property pack\"\n",
+    ")\n",
     "ax.set(xlabel=\"Node position\", ylabel=\"Osmotic pressure (Bar)\")\n",
     "plt.legend()\n",
     "\n",