From 73c5899e3d824a00c5abd23e5739251247629c9c Mon Sep 17 00:00:00 2001 From: Ben Dudson Date: Wed, 26 Nov 2025 15:41:03 -0800 Subject: [PATCH 1/2] bout.interpolate_parallel: Merge zeta chunking Before shifting using FFTs, merge zeta coordinate into one chunk so that the FFT can be performed. --- xbout/boutdataarray.py | 1 + 1 file changed, 1 insertion(+) diff --git a/xbout/boutdataarray.py b/xbout/boutdataarray.py index 3e54f402..71318e72 100644 --- a/xbout/boutdataarray.py +++ b/xbout/boutdataarray.py @@ -372,6 +372,7 @@ def interpolate_parallel( if not aligned_input: # Want output in non-aligned coordinates + da = da.chunk({'zeta':-1}) # One chunk in zeta for FFTs da = da.bout.from_field_aligned() if toroidal_points is not None and zcoord in da.sizes: From c5cd2ee9b33470161da2b7277b2dc1b11c3b8393 Mon Sep 17 00:00:00 2001 From: bendudson Date: Wed, 26 Nov 2025 23:44:24 +0000 Subject: [PATCH 2/2] Apply black formatting --- xbout/boutdataarray.py | 2 +- xbout/xarraybackend.py | 4 ++-- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/xbout/boutdataarray.py b/xbout/boutdataarray.py index 71318e72..d0b85a8c 100644 --- a/xbout/boutdataarray.py +++ b/xbout/boutdataarray.py @@ -372,7 +372,7 @@ def interpolate_parallel( if not aligned_input: # Want output in non-aligned coordinates - da = da.chunk({'zeta':-1}) # One chunk in zeta for FFTs + da = da.chunk({"zeta": -1}) # One chunk in zeta for FFTs da = da.bout.from_field_aligned() if toroidal_points is not None and zcoord in da.sizes: diff --git a/xbout/xarraybackend.py b/xbout/xarraybackend.py index 38f4d65c..6a9644c5 100644 --- a/xbout/xarraybackend.py +++ b/xbout/xarraybackend.py @@ -1,6 +1,6 @@ """License: - Distributed under the OSI-approved Apache License, Version 2.0. See - accompanying file Copyright.txt for details. +Distributed under the OSI-approved Apache License, Version 2.0. See +accompanying file Copyright.txt for details. """ from __future__ import annotations