There is an issue with the predict method in vector_auto_regression. When you use a model with an order higher than 1, the coefficients can not reshape to square matrices. However, in the predict method, it uses the ‘dense’ to get the data of connectivity output, which is not possible.