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