fix enumeration tutorial typos (#3109)

pyro-ppl · Jun 26, 2022 · 30c803b · 30c803b
1 parent f4b13ec
commit 30c803b
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/tutorial/source/enumeration.ipynb b/tutorial/source/enumeration.ipynb
@@ -527,9 +527,9 @@
     "    plate_1 = pyro.plate(\"plate_1\", 10, dim=-1)  # vectorized\n",
     "    plate_2 = pyro.plate(\"plate_2\", 10, dim=-2)  # vectorized\n",
     "    with plate_1:\n",
-    "        x = pyro.sample(\"y\", dist.Bernoulli(0.5))\n",
+    "        x = pyro.sample(\"x\", dist.Bernoulli(0.5))\n",
     "    with plate_2:\n",
-    "        y = pyro.sample(\"x\", dist.Bernoulli(0.5))\n",
+    "        y = pyro.sample(\"y\", dist.Bernoulli(0.5))\n",
     "    with plate_1, plate2:\n",
     "        z = pyro.sample(\"z\", dist.Bernoulli((1. + x + y) / 4.))\n",
     "        ...\n",
@@ -543,9 +543,9 @@
     "    plate_1 = pyro.plate(\"plate_1\", 10, dim=-1)  # vectorized\n",
     "    plate_2 = pyro.plate(\"plate_2\", 10)          # sequential\n",
     "    with plate_1:\n",
-    "        x = pyro.sample(\"y\", dist.Bernoulli(0.5))\n",
+    "        x = pyro.sample(\"x\", dist.Bernoulli(0.5))\n",
     "    for i in plate_2:\n",
-    "        y = pyro.sample(f\"x_{i}\", dist.Bernoulli(0.5))\n",
+    "        y = pyro.sample(f\"y_{i}\", dist.Bernoulli(0.5))\n",
     "        with plate_1:\n",
     "            z = pyro.sample(f\"z_{i}\", dist.Bernoulli((1. + x + y) / 4.))\n",
     "            ...\n",