import autoarray as aa
import autoarray.plot as aplt
import numpy as np

grid = aa.grid.uniform(shape_2d=(11, 11), pixel_scales=1.0)
color_array = np.linspace(start=0.0, stop=1.0, num=grid.shape_1d)

aplt.grid(grid=grid, color_array=color_array)
