Fix: OverflowError error when casting approx sum to integer

CartoDB · Jan 8, 2025 · ec74108 · ec74108
1 parent f9f5ff5
commit ec74108
Showing 1 changed file with 8 additions and 2 deletions.
diff --git a/raster_loader/io/common.py b/raster_loader/io/common.py
@@ -507,8 +507,14 @@ def raster_band_approx_stats(
     _sum = 0
     sum_squares = 0
     if count > 0:
-        _sum = int(np.sum(samples_band))
-        sum_squares = int(np.sum(np.array(samples_band) ** 2))
+        try:
+            _sum = int(np.sum(samples_band))
+        except OverflowError:
+            _sum = float("inf")
+        try:
+            sum_squares = int(np.sum(np.array(samples_band) ** 2))
+        except OverflowError:
+            sum_squares = float("inf")
 
     if basic_stats:
         quantiles = None