Re-add highlight of pattern in feature maps tutorial

Signed-off-by: Håkon Wiik Ånes <hwaanes@gmail.com>
pyxem · May 28, 2023 · 563ff5b · 563ff5b
1 parent 1ccf51c
commit 563ff5b
Showing 1 changed file with 5 additions and 3 deletions.
diff --git a/doc/tutorials/feature_maps.ipynb b/doc/tutorials/feature_maps.ipynb
@@ -398,13 +398,15 @@
    "metadata": {},
    "outputs": [],
    "source": [
-    "s.xmap.plot(\n",
+    "fig = s.xmap.plot(\n",
     "    adp3.ravel(),\n",
     "    cmap=\"gray\",\n",
     "    colorbar=True,\n",
     "    colorbar_label=\"Average dot product\",\n",
     "    remove_padding=True,\n",
-    ")"
+    "    return_figure=True\n",
+    ")\n",
+    "fig.axes[0].scatter(x, y, c=\"r\");"
    ]
   }
  ],
@@ -424,7 +426,7 @@
    "name": "python",
    "nbconvert_exporter": "python",
    "pygments_lexer": "ipython3",
-   "version": "3.10.9"
+   "version": "3.10.11"
   }
  },
  "nbformat": 4,