def seir_f(t, y, beta, sigma, gamma): s, e, i, r = y return np.array([-beta * i * s, -sigma * e + beta * i * s, -gamma * i + sigma * e, gamma * i])