Forward method inconsistent for time series transformer

Intermediate
1

Hello

According to the declaration of the forward method of TimeSeriesTransformerForPrediction the returned values can be either a tuple or an instance of Seq2SeqTSModelOutput class.

However the code returns an instance of Seq2SeqTSPredictionOutput class, which includes the distribution and its parameters.

I believe that the definition should be updated according to the actual returned object type.

Regards

David