Forward method inconsistent for time series transformer
HelloAccording 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.RegardsDavid