hotfix: call cholesky instead of root decomposition for unbroadcasted_scale_tril
@@ -47,8 +47,7 @@ class _MultivariateNormalBase(TMultivariateNormal, Distribution):
def _unbroadcasted_scale_tril(self):
if self.islazy and self.__unbroadcasted_scale_tril is None:
# cache root decoposition
- with settings.fast_computations(covar_root_decomposition=False):
- ust = self.lazy_covariance_matrix.root_decomposition().root.evaluate()
+ ust = self.lazy_covariance_matrix.cholesky()
self.__unbroadcasted_scale_tril = ust
return self.__unbroadcasted_scale_tril
