Coverage for causalspyne/wishart.py: 100%

1import numpy as np 

2from scipy import stats 

3 

4 

5def gen_weight_matrix(rng, n=4): 

6 df = n + 1 

7 scale_matrix = np.eye(n) 

8 weight = stats.wishart(df=df, scale=scale_matrix, seed=rng).rvs(size=1) 

9 if weight.shape == (): 

10 weight = weight[None, None] 

11 return weight