Could you add some docs to the class docstring for the `grid_shape`, I think the NumPy style it would be like:

```
Attributes
----------
grid_shape : ???
```