Time Series Transformer Python