fd3a6fbaa2ca0b7b1fb70071fec33d7a4cebd14d,gpytorch/lazy/sum_lazy_tensor.py,SumLazyTensor,diag,#SumLazyTensor#,79

Before Change

            raise AttributeError("other must be a LazyTensor")

    def diag(self):
        diags = [lazy_tensor.diag().contiguous() for lazy_tensor in self.lazy_tensors]
        size = diags[0].size()
        res = sum(diag.view(-1) for diag in diags)
        res = res.view(size)
        return res

After Change

            raise AttributeError("other must be a LazyTensor")

    def diag(self):
        return sum(lazy_tensor.diag().contiguous() for lazy_tensor in self.lazy_tensors)

Instances