fixing changes to fft related PR

PMEAL · Aug 24, 2024 · 26c48ad · 26c48ad
1 parent a923fc6
commit 26c48ad
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/src/porespy/metrics/_funcs.py b/src/porespy/metrics/_funcs.py
@@ -770,9 +770,9 @@ def _radial_profile(autocorr, bins, pf=None, voxel_size=1):
 def _get_radial_sum(dt, bins, bin_size, autocorr):
     radial_sum = np.zeros_like(bins[:-1], dtype=np.float64)
     for i, r in enumerate(bins[:-1]):
-        mask = (dt <= r) & (dt > (r - bin_size[i])) 
-        if np.any(mask):
-            radial_sum[i] = np.sum(autocorr[mask], dtype=np.float64) / np.sum(mask, dtype=np.float64)
+        mask = (dt <= r) * (dt > (r - bin_size[i]))
+        radial_sum[i] = (np.sum(np.ravel(autocorr)[np.ravel(mask)], dtype=np.int64) /
+                         np.sum(mask, dtype=np.int64))
     return radial_sum