fix nans

ferrine · ferrine · commit f83656f877bf · 2023-06-05T16:55:04.000+03:00
diff --git a/pymc_experimental/distributions/multivariate/r2d2m2cp.py b/pymc_experimental/distributions/multivariate/r2d2m2cp.py
@@ -29,7 +29,7 @@ def _psivar2musigma(psi: pt.TensorVariable, explained_var: pt.TensorVariable, ps
     mu = sigma * pi * 2**0.5
     if psi_mask is not None:
         return (
-            pt.where(psi_mask, mu, pt.sign(mu) * explained_var**0.5),
+            pt.where(psi_mask, mu, pt.sign(pi) * explained_var**0.5),
             pt.where(psi_mask, sigma, 0),
         )
     else:

Original file line number	Diff line number	Diff line change
`@@ -29,7 +29,7 @@ def _psivar2musigma(psi: pt.TensorVariable, explained_var: pt.TensorVariable, ps`
`29`	`29`	`mu = sigma * pi * 2**0.5`
`30`	`30`	`if psi_mask is not None:`
`31`	`31`	`return (`
`32`		`- pt.where(psi_mask, mu, pt.sign(mu) * explained_var**0.5),`
	`32`	`+ pt.where(psi_mask, mu, pt.sign(pi) * explained_var**0.5),`
`33`	`33`	`pt.where(psi_mask, sigma, 0),`
`34`	`34`	`)`
`35`	`35`	`else:`