def set_axis_limits(axis_limits):
if axis_limits is not None:<EOL><INDENT>plt.axis(axis_limits)<EOL><DEDENT>
Set the axis limits of the figure the grid is plotted on.

    Parameters
    -----------
    axis_limits : []
        The axis limits of the figure on which the grid is plotted, following [xmin, xmax, ymin, ymax].