Hi, I'm following the [Time Series Generation with VAEs](https://github.com/AlexanderVNikitin/tsgm/blob/main/tutorials/VAEs/VAE.ipynb) tutorial along with the [Lotka Volterra](https://tsgm.readthedocs.io/en/latest/_modules/tsgm/simulator.html#LotkaVolterraSimulator) documentation and it won't work. Is the documenation different from the tutorials? Please advise. Thank you, Richard