diff --git a/modules/icing/pirep_goes.py b/modules/icing/pirep_goes.py index e8bd0c97659d89db6e416bcac31dd8462c573a44..21a29168102f5a5b1b3b8fe19c67993c7c7a00bf 100644 --- a/modules/icing/pirep_goes.py +++ b/modules/icing/pirep_goes.py @@ -1291,13 +1291,18 @@ def normalize(data, param, mean_std_dict): if mean_std_dict.get(param) is None: return data + shape = data.shape + data = data.flatten() + mean, std = mean_std_dict.get(param) data -= mean data /= std - not_valid = np.invert(np.isnan(data)) + not_valid = np.isnan(data) data[not_valid] = 0 + data = np.reshape(data, shape) + return data