Изменение пределов осей графа не работает - python 3, matplolib

Question

Я пытаюсь создать анимированный график функции. При использовании графа FuncAnimation границы не меняются. Вот мой код:

import matplotlib.pyplot as plt
import numpy as np
import matplotlib.animation as animation

fig = plt.figure()
ax = plt.axes()

ln, = ax.plot([],[])

def init():
    x = np.linspace(0, 10, 100)
    y = x ** 2

    ln.set_data(x, y)
    ax.set_xlim(0, 10)
    ax.set_ylim(0, 100)

    return ln,

def update(i):
    x = np.linspace(1 + i, 1 + 10*i, 100)
    y = x ** 2

    currentDomainMax = int(x[-1])
    currentRangeMax = int(y[-1])

    ax.set_xlim(0, currentDomainMax)
    ax.set_ylim(0, currentRangeMax)
    ln.set_data(x, y)

    return ln,

ani = animation.FuncAnimation(fig, init_func=init, func=update, frames=10, interval=100, blit=True, repeat=False)
plt.show()

График отображается, но ограничения не меняются на протяжении всей анимации. Не могу понять, почему, я новичок в анимации в python и matplotlib, любая помощь приветствуется.