parallax plotting bug

py4dstem · Nov 14, 2023 · 6a9f1ca · 6a9f1ca
1 parent 53ac42f
commit 6a9f1ca
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/py4DSTEM/process/phase/iterative_parallax.py b/py4DSTEM/process/phase/iterative_parallax.py
@@ -603,7 +603,9 @@ def preprocess(
                 0.5 * (self._reciprocal_sampling[0] * self._dp_mask.shape[0]),
                 -0.5 * (self._reciprocal_sampling[0] * self._dp_mask.shape[0]),
             ]
-            ax[1].imshow(self._dp_mask, extent=reciprocal_extent, cmap="gray")
+            ax[1].imshow(
+                self._asnumpy(self._dp_mask), extent=reciprocal_extent, cmap="gray"
+            )
             ax[1].set_title("DP mask")
             ax[1].set_ylabel(r"$k_x$ [$A^{-1}$]")
             ax[1].set_xlabel(r"$k_y$ [$A^{-1}$]")