diff --git a/tests/create_test_data.py b/tests/create_test_data.py index 6a307d5..2f9d982 100644 --- a/tests/create_test_data.py +++ b/tests/create_test_data.py @@ -36,9 +36,9 @@ def _arr(value, chunks, order): arr = arr + noise if land_mask: - arr[ - multi_indices[0], multi_indices[1], : - ] = np.nan # add NaNs to mimic land mask + arr[multi_indices[0], multi_indices[1], :] = ( + np.nan + ) # add NaNs to mimic land mask if use_xr: arr = xr.DataArray(arr) if chunks: