src/transformers/time_series_utils.py/AffineTransformed/mean
class AffineTransformed:    def mean(self):
        """
        Returns the mean of the distribution.
        """
        return self.base_dist.mean * self.scale + self.loc