Fix

Signed-off-by: Eric Kerfoot <[email protected]>
Project-MONAI · Sep 6, 2024 · 18eee5d · 18eee5d
2 parents 1e4f353 + ba976a5
commit 18eee5d
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/generation/2d_vqgan/2d_vqgan_tutorial.ipynb b/generation/2d_vqgan/2d_vqgan_tutorial.ipynb
@@ -626,6 +626,7 @@
     "fig, ax = plt.subplots(nrows=num_inters, ncols=1, sharey=True)\n",
     "ax=ensure_tuple(ax)\n",
     "\n",
+    "ax = ensure_tuple(ax)\n",
     "for image_n in range(num_inters):\n",
     "    reconstructions = torch.reshape(intermediary_images[image_n], (64 * n_example_images, 64)).T\n",
     "    ax[image_n].imshow(reconstructions.cpu(), cmap=\"gray\")\n",