Fixed reconciling experts smoke test.

pyro-ppl · Dec 31, 2024 · 8c4cd84 · 8c4cd84
1 parent 200fd36
commit 8c4cd84
Showing 1 changed file with 4 additions and 4 deletions.
diff --git 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()"