From 3060594b8fa0acb48afe97064e2f4461a6696111 Mon Sep 17 00:00:00 2001 From: David Horsley Date: Wed, 20 Dec 2023 10:23:30 +1100 Subject: [PATCH] use ive for numerical stability --- pymc/gp/cov.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pymc/gp/cov.py b/pymc/gp/cov.py index 9b04e23977..0eddaea932 100644 --- a/pymc/gp/cov.py +++ b/pymc/gp/cov.py @@ -810,7 +810,7 @@ def power_spectral_density_approx(self, J: TensorLike) -> TensorVariable: a = 1 / pt.square(self.ls) c = pt.where(J > 0, 2, 1) - q2 = c * pt.iv(J, a) / pt.exp(a) + q2 = c * pt.ive(J, a) return q2