There is a small labeling bug in Autoregressive
: the parameter dimensions are using AR_PARAM_DIM
(always equal to "lag_ar"
), which is inconsistent for multivariate param dims. I changed it to f"lag_{self.name}"
(e.g "lag_auto_regressive"
) and will push the PR as soon as I can