Не работает визуализация

import numpy as np
import matplotlib.pyplot as plt
import time
from IPython.display import display, clear_output
 
%matplotlib inline
 
# Параметры
b_start = 1.88
b_end = 2.00
step = 0.01
delay = 1.0
 
D = 1.5
N = 5
t_start = 3.4
t_end = 6.8
num_points = 10000 # Уменьшим для скорости
 
t = np.linspace(t_start, t_end, num_points)
 
def weierstrass(t, b, D, N):
W = np.zeros_like(t)
for n in range(-N, N+1):
W += (1 - np.cos(b**n * t)) / (b**((2 - D) * n))
return W
 
# Создаем фигуру один раз
fig, ax = plt.subplots(figsize=(15, 5))
plt.close(fig) # Закрываем, чтобы не показывалась лишний раз
 
current_b = b_start
while current_b <= b_end:
W = weierstrass(t, current_b, D, N)
 
# Очищаем вывод и рисуем заново
clear_output(wait=True)
fig, ax = plt.subplots(figsize=(15, 5))
ax.plot(t, W, 'b-', linewidth=1)
ax.set_title(f'Функция Вейерштрасса-Мандельброта (b = {current_b:.2f})')
ax.set_xlabel('Время')
ax.set_ylabel('Значение')
ax.grid(True)
ax.set_xlim(t_start, t_end)
ax.set_ylim(-5, 5)
 
display(fig) # Отображаем график
plt.close(fig) # Закрываем, чтобы не накапливались в памяти
 
time.sleep(delay)
current_b = round(current_b + step, 2)
 
print("Анимация завершена!")

def weierstrass(t, b, D, N):
    W = np.zeros_like(t)
    for n in range(-N, N+1):
        W += (1 - np.cos(b**n * t)) / (b**((2 - D) * n))
    return W