From 52ea4922b00f54352c597a991042c12f3c751539 Mon Sep 17 00:00:00 2001
From: smribet <stephanie.ribet@gmail.com>
Date: Sun, 5 Jan 2025 06:31:48 -0800
Subject: [PATCH] swap reciprocal space too

---
 py4DSTEM/tomography/tomography.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/py4DSTEM/tomography/tomography.py b/py4DSTEM/tomography/tomography.py
index 0ba96ddfe..cddf99b1a 100644
--- a/py4DSTEM/tomography/tomography.py
+++ b/py4DSTEM/tomography/tomography.py
@@ -544,6 +544,8 @@ def _prepare_datacube(
                         0,
                         np.inf,
                     )
+        if self._transpose_xy:
+            datacube.data = datacube.data.swapaxes(-1, -2)
 
         # resize diffraction space
         if diffraction_intensities_shape is not None: