Merge pull request #980 from cmaloney111/dev

Avoid overflow errors with _get_radial_sum #maint #bug
PMEAL · Aug 22, 2024 · d6146fc · d6146fc
2 parents 4b3b632 + 6bbac76
commit d6146fc
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/src/porespy/__version__.py b/src/porespy/__version__.py
@@ -1 +1 @@
-__version__ = '3.0.0a0.dev13'
+__version__ = '3.0.0a0.dev14'
diff --git a/src/porespy/metrics/_funcs.py b/src/porespy/metrics/_funcs.py
@@ -768,11 +768,11 @@ def _radial_profile(autocorr, bins, pf=None, voxel_size=1):
 
 @njit(parallel=False)
 def _get_radial_sum(dt, bins, bin_size, autocorr):
-    radial_sum = np.zeros_like(bins[:-1])
+    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]))
-        radial_sum[i] = np.sum(np.ravel(autocorr)[np.ravel(mask)], dtype=np.int64) \
-            / np.sum(mask)
+        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)
     return radial_sum
Original file line number	Diff line number	Diff line change
		@@ -1 +1 @@
		__version__ = '3.0.0a0.dev13'
		__version__ = '3.0.0a0.dev14'