From 6a9f1cae8ee982455635d3894ceda04c61e6b7b3 Mon Sep 17 00:00:00 2001 From: Stephanie Ribet Date: Mon, 13 Nov 2023 17:26:57 -0800 Subject: [PATCH] parallax plotting bug --- py4DSTEM/process/phase/iterative_parallax.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/py4DSTEM/process/phase/iterative_parallax.py b/py4DSTEM/process/phase/iterative_parallax.py index 716e1d782..04366b8c2 100644 --- 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}$]")