Matplotlib animations in VSCode jupyter notebook

anim.py

# %%
%matplotlib ipympl

# %%
import numpy as np
from matplotlib import pyplot as plt
from matplotlib.animation import FuncAnimation

# %%
fig, ax = plt.subplots()

line, = ax.plot([])

ax.set_xlim(0, 2*np.pi)
ax.set_ylim(-1.1, 1.1)

def animate(frame):
    x = np.linspace(0, 2*np.pi, 100)
    y = np.sin(x + 2*np.pi * frame / 100)
    line.set_data((x, y))
    return line

anim = FuncAnimation(fig, animate, frames=100, interval=20)

plt.show()