Fixes type casting when data are None

MoseleyS · Sep 11, 2023 · d5096ed · d5096ed
1 parent b2b211d
commit d5096ed
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/improver/nbhood/nbhood.py b/improver/nbhood/nbhood.py
@@ -316,11 +316,12 @@ def _calculate_neighbourhood(
         data[data_mask] = 0
 
         if self.sum_only:
-            area_sum = None
+            max_extreme_data = None
         else:
             area_sum = self._do_nbhood_sum(valid_data_mask)
+            max_extreme_data = area_sum.astype(loc_data_dtype)
         # Where data are all ones in nbhood, result will be same as area_sum
-        data = self._do_nbhood_sum(data, max_extreme=area_sum.astype(loc_data_dtype))
+        data = self._do_nbhood_sum(data, max_extreme=max_extreme_data)
 
         if not self.sum_only:
             with np.errstate(divide="ignore", invalid="ignore"):