анимация не работает в блокноте jupyter для библиотеки dgl

Question

Я пытаюсь получить результат анимации из библиотеки dgl.

ani = animation.FuncAnimation(fig, draw, frames=len(all_logits), interval=200)

Я не могу получить график в виде вывода в jupyter. он вызывает ошибку, когда я пытаюсь использовать метод ani.to_save() или to_html().