From 8c4cd84950dfeb77eda981148dc19fe3e1ef6f0d Mon Sep 17 00:00:00 2001
From: Ben Zickel <benzickel@gmail.com>
Date: Tue, 31 Dec 2024 17:32:38 +0200
Subject: [PATCH] Fixed reconciling experts smoke test.

---
 tutorial/source/reconciling_experts.ipynb | 8 ++++----
 1 file changed, 4 insertions(+), 4 deletions(-)

diff --git a/tutorial/source/reconciling_experts.ipynb b/tutorial/source/reconciling_experts.ipynb
index 2a9f63b2ae..57eb515859 100644
--- a/tutorial/source/reconciling_experts.ipynb
+++ b/tutorial/source/reconciling_experts.ipynb
@@ -44,8 +44,8 @@
     "from matplotlib import pyplot as plt\n",
     "\n",
     "smoke_test = ('CI' in os.environ)  # for use in continuous integration testing\n",
-    "num_samples = 2 if smoke_test else 1000\n",
-    "num_svi_iter = 2 if smoke_test else 1001"
+    "num_samples = 4 if smoke_test else 1000\n",
+    "num_svi_iter = 4 if smoke_test else 1001"
    ]
   },
   {
@@ -243,7 +243,7 @@
        "</svg>\n"
       ],
       "text/plain": [
-       "<graphviz.graphs.Digraph at 0x24445b0f8f0>"
+       "<graphviz.graphs.Digraph at 0x2417c3f7a10>"
       ]
      },
      "execution_count": 6,
@@ -622,7 +622,7 @@
    ],
    "source": [
     "plt.figure()\n",
-    "for i in range(1000):\n",
+    "for i in range(new_samples.shape[0]):\n",
     "    plt.plot(new_samples[i, ..., 0].detach().numpy(), alpha=0.05, color='blue')\n",
     "    plt.plot(new_samples[i, ..., 1].detach().numpy(), alpha=0.05, color='orange')\n",
     "plt.grid()"