fix checks

arviz-devs · Aug 2, 2022 · 3d7a512 · 3d7a512
1 parent 69593cb
commit 3d7a512
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/xarray_einstats/stats.py b/src/xarray_einstats/stats.py
@@ -5,8 +5,8 @@
 
 import numpy as np
 import xarray as xr
-from scipy import stats
 from numpy.linalg import LinAlgError
+from scipy import stats
 
 from .linalg import cholesky, eigh
 
@@ -369,7 +369,7 @@ def rvs(self, mean=None, cov=None, dims=None, *, size=1, rv_dims=None, random_st
         except LinAlgError:
             k = len(cov[dim1])
             eye = xr.DataArray(np.eye(k), dims=list(dims))
-            cov_chol = cholesky(cov + 1e-10*eye, dims=dims)
+            cov_chol = cholesky(cov + 1e-10 * eye, dims=dims)
         std_norm = XrContinuousRV(stats.norm, xr.zeros_like(mean.rename({dim1: dim2})), 1)
         samples = std_norm.rvs(size=size, dims=rv_dims, random_state=random_state)
         return mean + xr.dot(cov_chol, samples, dims=dim2)