local_group_arrays

[csskene]

Hi
I've noticed that local_group_arrays now returns an error. When running m, ell, *_ = d.coeff_layout.local_group_arrays(b.domain,scales=1) I get an error when the function tries to find the chunk_shape

 chunk_shape = domain.chunk_shape(self)

                  ^^^^^^^^^^^^^^^^^^

AttributeError: 'CachedMethod' object has no attribute 'chunk_shape'

Attached is a small example that exhibits this problem
Thanks,
Calum
group_arrays.txt

[csskene]

Just realised that this issue is due to changes in Dedalus. My example can be fixed by changing

m, ell, *_ = d.coeff_layout.local_group_arrays(b.domain,scales=1)

to

m, ell, *_ = d.coeff_layout.local_group_arrays(b.domain(d), scales=1)

This fixes things as now b.domain is a method that takes the distributor as an argument.