We should explicitly describe the size of the grid_shape here:

"The desired grid shape for the montage as `(alpha_y, alpha_x)` ..."

?