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

Question

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

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

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