From 5f5e1be9376f02b56535505a96dcc612f3ec33a5 Mon Sep 17 00:00:00 2001 From: Jason Boutte Date: Wed, 8 Nov 2023 15:21:45 -0800 Subject: [PATCH] Fixes wrapping when using np.take --- xcdat/regridder/regrid2.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/xcdat/regridder/regrid2.py b/xcdat/regridder/regrid2.py index 1a0c351e..1158bf05 100644 --- a/xcdat/regridder/regrid2.py +++ b/xcdat/regridder/regrid2.py @@ -130,10 +130,10 @@ def _regrid( # need to optimize for y in range(y_length): - y_seg = np.take(input_data, lat_mapping[y], axis=y_index) + y_seg = np.take(input_data, lat_mapping[y], axis=y_index, mode="wrap") for x in range(x_length): - x_seg = np.take(y_seg, lon_mapping[x], axis=x_index, mode="clip") + x_seg = np.take(y_seg, lon_mapping[x], axis=x_index, mode="wrap") cell_weight = np.dot( lat_weights[y].reshape((-1, 1)), lon_weights[x].reshape((1, -1))