Minor cosmetic PEP8 change: Remove space before: `grid_shape = (3, 4)` => ` grid_shape=(3, 4)`