fixes pymc-devs#3051

Ahanmr · Mar 18, 2020 · 9df27b3 · 9df27b3
1 parent 7420495
commit 9df27b3
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pymc3/distributions/multivariate.py b/pymc3/distributions/multivariate.py
@@ -327,7 +327,7 @@ def logp(self, value):
         TensorVariable
         """
         quaddist, logdet, ok = self._quaddist(value)
-        k = intX(value.shape[-1]).astype(theano.config.floatX)
+        k = floatX(value.shape[-1])
         norm = - 0.5 * k * pm.floatX(np.log(2 * np.pi))
         return bound(norm - 0.5 * quaddist - logdet, ok)
 
@@ -441,7 +441,7 @@ def logp(self, value):
         TensorVariable
         """
         quaddist, logdet, ok = self._quaddist(value)
-        k = intX(value.shape[-1]).astype(theano.config.floatX)
+        k = floatX(value.shape[-1])
 
         norm = (gammaln((self.nu + k) / 2.)
                 - gammaln(self.nu / 2.)