Format notebooks

iiasa · Jun 9, 2022 · 02e2775 · 02e2775
1 parent f083f73
commit 02e2775
Show file tree

Hide file tree

Showing 3 changed files with 21 additions and 25 deletions.
diff --git a/notebooks/run-example-ciceroscm.ipynb b/notebooks/run-example-ciceroscm.ipynb
@@ -128,9 +128,9 @@
     "model = \"ciceroscm\"\n",
     "model_version = \"v2019vCH4\"\n",
     "probabilistic_file = os.path.join(\n",
-    "    \"..\", \n",
+    "    \"..\",\n",
     "    \"data\",\n",
-    "    \"cicero\", \n",
+    "    \"cicero\",\n",
     "    \"subset_cscm_configfile.json\",\n",
     ")\n",
     "\n",

diff --git a/notebooks/run-example-custom.ipynb b/notebooks/run-example-custom.ipynb
@@ -112,12 +112,12 @@
     "    model=\"ciceroscm\",\n",
     "    model_version=\"v2019vCH4\",\n",
     "    probabilistic_file=os.path.join(\n",
-    "        \"..\", \n",
+    "        \"..\",\n",
     "        \"data\",\n",
-    "        \"cicero\", \n",
+    "        \"cicero\",\n",
     "        \"subset_cscm_configfile.json\",\n",
     "    ),\n",
-    "#     num_cfgs=600,\n",
+    "    #     num_cfgs=600,\n",
     "    num_cfgs=1,\n",
     "    test_run=False,\n",
     ")"
@@ -139,12 +139,8 @@
    "outputs": [],
    "source": [
     "# Where is the binary?\n",
-    "os.environ[\"MAGICC_EXECUTABLE_7\"]= os.path.join(\n",
-    "    \"path\",\n",
-    "    \"to\",\n",
-    "    \"magicc-v7.5.3\",\n",
-    "    \"bin\",\n",
-    "    \"magicc\"\n",
+    "os.environ[\"MAGICC_EXECUTABLE_7\"] = os.path.join(\n",
+    "    \"path\", \"to\", \"magicc-v7.5.3\", \"bin\", \"magicc\"\n",
     ")\n",
     "\n",
     "# How many MAGICC workers can run in parallel?\n",
@@ -162,7 +158,7 @@
     "    probabilistic_file=os.path.join(\n",
     "        \"path\",\n",
     "        \"to\",\n",
-    "        \"magicc-ar6-0fd0f62-f023edb-drawnset\", \n",
+    "        \"magicc-ar6-0fd0f62-f023edb-drawnset\",\n",
     "        \"0fd0f62-derived-metrics-id-f023edb-drawnset.json\",\n",
     "    ),\n",
     "    num_cfgs=600,\n",
@@ -238,7 +234,7 @@
    "source": [
     "idx = pd.MultiIndex.from_arrays(\n",
     "    [\n",
-    "        [\"stylised\"] * 4, \n",
+    "        [\"stylised\"] * 4,\n",
     "        [\"example\"] * 4,\n",
     "        [\n",
     "            \"Emissions|CO2|Energy and Industrial Processes\",\n",
@@ -248,13 +244,14 @@
     "        ],\n",
     "        [\"Mt CO2/yr\", \"Mt CO2/yr\", \"Mt CH4/yr\", \"kt N2O/yr\"],\n",
     "        [\"World\"] * 4,\n",
-    "    ], \n",
-    "    names=[\"model\", \"scenario\", \"variable\", \"unit\", \"region\"]\n",
+    "    ],\n",
+    "    names=[\"model\", \"scenario\", \"variable\", \"unit\", \"region\"],\n",
     ")\n",
     "years = np.arange(2010, 2100 + 1, 5)\n",
     "\n",
+    "\n",
     "def sigmoid(k, x, x0):\n",
-    "    return (1 / (1 + np.exp(-k * (x-x0))))\n",
+    "    return 1 / (1 + np.exp(-k * (x - x0)))\n",
     "\n",
     "\n",
     "base = pd.DataFrame(\n",
@@ -271,8 +268,10 @@
     "shuffle[\"scenario\"] = \"example_shuffle\"\n",
     "shuffle = shuffle.set_index(base.index.names)\n",
     "shuffle.loc[\n",
-    "    shuffle.index.get_level_values(\"variable\").isin([\"Emissions|CO2|Energy and Industrial Processes\"]),\n",
-    "    :\n",
+    "    shuffle.index.get_level_values(\"variable\").isin(\n",
+    "        [\"Emissions|CO2|Energy and Industrial Processes\"]\n",
+    "    ),\n",
+    "    :,\n",
     "] = 35000 * sigmoid(-1 / 5, years, 2040)\n",
     "\n",
     "inp = pyam.IamDataFrame(pd.concat([base, shuffle]))\n",

diff --git a/notebooks/run-example-magicc.ipynb b/notebooks/run-example-magicc.ipynb
@@ -174,11 +174,8 @@
    "outputs": [],
    "source": [
     "# Where is the binary\n",
-    "os.environ[\"MAGICC_EXECUTABLE_7\"]= os.path.join(\n",
-    "    os.environ[\"MAGICC_ROOT_FILES_DIR\"],\n",
-    "    \"magicc-v7.5.3\",\n",
-    "    \"bin\",\n",
-    "    \"magicc\"\n",
+    "os.environ[\"MAGICC_EXECUTABLE_7\"] = os.path.join(\n",
+    "    os.environ[\"MAGICC_ROOT_FILES_DIR\"], \"magicc-v7.5.3\", \"bin\", \"magicc\"\n",
     ")\n",
     "\n",
     "# How many MAGICC workers can run in parallel?\n",
@@ -209,8 +206,8 @@
     "model = \"magicc\"\n",
     "model_version = \"v7.5.3\"\n",
     "probabilistic_file = os.path.join(\n",
-    "    os.environ[\"MAGICC_ROOT_FILES_DIR\"], \n",
-    "    \"magicc-ar6-0fd0f62-f023edb-drawnset\", \n",
+    "    os.environ[\"MAGICC_ROOT_FILES_DIR\"],\n",
+    "    \"magicc-ar6-0fd0f62-f023edb-drawnset\",\n",
     "    \"0fd0f62-derived-metrics-id-f023edb-drawnset.json\",\n",
     ")\n",
     "\n",