From 60b03527f6154e09ef755de8513256d6e2bc904f Mon Sep 17 00:00:00 2001
From: Samuel Burbulla <s.burbulla@appliedai-institute.de>
Date: Tue, 5 Dec 2023 09:04:38 +0200
Subject: [PATCH] Update notebook.

---
 notebooks/sine.ipynb | 15 +++------------
 1 file changed, 3 insertions(+), 12 deletions(-)

diff --git a/notebooks/sine.ipynb b/notebooks/sine.ipynb
index 10ac28b4..66653dc0 100644
--- a/notebooks/sine.ipynb
+++ b/notebooks/sine.ipynb
@@ -6,8 +6,8 @@
    "source": [
     "# Sine Example\n",
     "\n",
-    "This example shows how to use a neural operator to learn the sine function in\n",
-    "a self-supervised manner."
+    "This example trains a neural operator on the sine function in a self-supervised \n",
+    "manner."
    ]
   },
   {
@@ -116,9 +116,7 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "fig, axs = plt.subplots(size // 4, 4, figsize=(16, 3 * size // 4))\n",
-    "if size // 4 == 1:\n",
-    "    axs = [axs]\n",
+    "fig, axs = plt.subplots(2, 4, figsize=(16, 6))\n",
     "\n",
     "for i in range(size):\n",
     "    ax = axs[i // 4][i % 4]\n",
@@ -145,13 +143,6 @@
     "plot_evaluation(model, dataset, obs, ax=ax)\n",
     "plot_observation(obs, ax=ax)"
    ]
-  },
-  {
-   "cell_type": "code",
-   "execution_count": null,
-   "metadata": {},
-   "outputs": [],
-   "source": []
   }
  ],
  "metadata": {