From 6c48895acad8fe294fbbda801cc5d98e00279ff8 Mon Sep 17 00:00:00 2001
From: avdudchenko <33663878+avdudchenko@users.noreply.github.com>
Date: Thu, 12 Sep 2024 12:29:26 -0700
Subject: [PATCH] Update integration_with_ro.ipynb

---
 .../tutorials/integration_with_ro.ipynb       | 19 ++++++++++++-------
 1 file changed, 12 insertions(+), 7 deletions(-)

diff --git a/src/reaktoro_pse/tutorials/integration_with_ro.ipynb b/src/reaktoro_pse/tutorials/integration_with_ro.ipynb
index e8a1f89..97609ee 100644
--- 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",