From fafec622f87fb30a99da3499e769fe416836eec0 Mon Sep 17 00:00:00 2001
From: Samuel St-Jean <3030760+samuelstjean@users.noreply.github.com>
Date: Mon, 22 Apr 2024 15:52:57 -0400
Subject: [PATCH] float32 output stuff

---
 autodmri/estimator.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/autodmri/estimator.py b/autodmri/estimator.py
index 1239aef..2821610 100644
--- a/autodmri/estimator.py
+++ b/autodmri/estimator.py
@@ -81,8 +81,8 @@ def estimate_from_dwis(data, axis=-2, return_mask=False, exclude_mask=None, ncor
     output = Parallel(n_jobs=ncores)(delayed(_inner)(swapped_data[i], median, exclude_mask[i], method) for i in ranger)
 
     # output is each slice we took along axis, so the mask might be reversed
-    sigma = np.zeros(len(output))
-    N = np.zeros(len(output))
+    sigma = np.zeros(len(output), dtype=np.float32)
+    N = np.zeros(len(output), dtype=np.float32)
     mask = np.zeros(data.shape[:-1], dtype=np.int16).swapaxes(0, axis)
 
     for i, s in enumerate(output):