From d0d749f8d92d6e7aaaf0c4a55a025fd086156ed1 Mon Sep 17 00:00:00 2001 From: Javier Vegas-Regidor Date: Fri, 4 Jun 2021 14:02:27 +0200 Subject: [PATCH] Use nansum --- diagonals/ohc.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/diagonals/ohc.py b/diagonals/ohc.py index fb83754..d08f26d 100644 --- a/diagonals/ohc.py +++ b/diagonals/ohc.py @@ -163,7 +163,7 @@ def _compute_ohc_cpu(layers, thetao, weights, area): """ ohc = [] for layer in range(len(layers)): - ohc_layer = da.sum( + ohc_layer = da.nansum( _multiply_array(thetao, weights[layer]), axis=1 ) @@ -171,7 +171,7 @@ def _compute_ohc_cpu(layers, thetao, weights, area): ohc1d_total = [] for i, basin in enumerate(area): ohc_basin = _multiply_array(ohc_layer, area[basin]) - ohc1d = da.sum(ohc_basin, axis=(1, 2)) + ohc1d = da.nansum(ohc_basin, axis=(1, 2)) ohc1d_total.append(ohc1d) return ohc, ohc1d_total -- GitLab