From e9e2fd4beef36acc7b01beba2f091314a5d2d000 Mon Sep 17 00:00:00 2001
From: allevitan <allevitan@gmail.com>
Date: Sun, 13 Oct 2024 13:14:19 -0400
Subject: [PATCH] Get all tests passing

---
 src/cdtools/tools/initializers/initializers.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/cdtools/tools/initializers/initializers.py b/src/cdtools/tools/initializers/initializers.py
index c0385ee..7e727af 100644
--- a/src/cdtools/tools/initializers/initializers.py
+++ b/src/cdtools/tools/initializers/initializers.py
@@ -57,7 +57,7 @@ def exit_wave_geometry(det_basis, det_shape, wavelength, distance, oversampling=
 
     # This method should work for a general parallelogram-shaped detector
     det_shape = det_basis * det_shape.to(t.float32)
-    pinv_basis = t.tensor(np.linalg.pinv(det_shape).transpose()).to(t.float32)
+    pinv_basis = t.linalg.pinv(det_shape).transpose(0,1)
     real_space_basis = pinv_basis * wavelength * distance
 
     return real_space_basis