text: 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)
  