---

def pyass(nx, nt, hx, ht, f, u):
    U = u.copy()
    # Если параметры фиксированные, лучше задавать их по умолчанию или передавать при вызове:
    nx, nt = 5, 3
    hx, ht = 1/4, 1/2
    f = 2
    u = [[1 for j in range(nt)] for i in range(nx)]
    
    for i in range(1, nx-1):
        for j in range(1, nt-1):
            u[i][j] = (((u[i-1][j] + u[j+1][j]) / hx**2 + u[i][j-1] + u[i][j+1]) / ht**2 - f) / (2/hx**2 + 2/ht**2)
            
    return U  # <- Отступ должен соответствовать телу функции