glass-dev / glass

Generator for Large Scale Structure
https://glass.readthedocs.io
MIT License
31 stars 7 forks source link

BUG(shells): fix partition() for multi-dimensional arrays #132

Closed ntessore closed 11 months ago

ntessore commented 11 months ago

Fix the partition() function for multi-dimensional arrays. The axis corresponding to the shells is now the first one.

Fixed: partition() now works correctly with functions having extra axes. Changed: The output of partition() now has the shells axis as its first.