This message was deleted.

Let me have a look at it

I went back to this and fixed the issue, here is a PR: https://github.com/Nixtla/nixtla/pull/44 I will keep working on reproducing the colab notebook and let you know how it goes (and possibly ask more questions if I get stuck again) simple smile

Thank you very much pietro. We have been very busy Planning a new release and haven't been able to check this but @fede (nixtla) (they/them) will take a look at it next week.