Skip to content

Add xr.Dataset support to rechunk_no_shuffle#1070

Merged
brendancol merged 3 commits intoissue-1067from
issue-1069
Mar 24, 2026
Merged

Add xr.Dataset support to rechunk_no_shuffle#1070
brendancol merged 3 commits intoissue-1067from
issue-1069

Conversation

@brendancol
Copy link
Contributor

Summary

  • rechunk_no_shuffle now accepts xr.Dataset in addition to xr.DataArray
  • For Datasets, each dask-backed variable is rechunked independently; numpy variables pass through
  • Added rechunk_no_shuffle to the Dataset .xrs accessor
  • Updated notebook with Dataset examples

Test plan

  • 5 new Dataset tests (rechunks dask vars, preserves numpy vars, preserves attrs/coords, preserves values, Dataset accessor)
  • Updated type-error test for new accepted types
  • Dataset accessor added to expected-methods check
  • All 40 tests pass

Closes #1069

Accepts both DataArray and Dataset. For Datasets, each dask-backed
variable is rechunked independently. Also adds the method to the
Dataset .xrs accessor.
@github-actions github-actions bot added the performance PR touches performance-sensitive code label Mar 24, 2026
@brendancol brendancol merged commit df8cfce into issue-1067 Mar 24, 2026
11 checks passed
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

performance PR touches performance-sensitive code

Projects

None yet

Development

Successfully merging this pull request may close these issues.

1 participant