Fix KroneckerNormal ndim_supp

Author: ricardoV94

pymc/distributions/multivariate.py

@@ -1829,14 +1829,11 @@ def logp(value, mu, rowchol, colchol):
 
 class KroneckerNormalRV(RandomVariable):
     name = "kroneckernormal"
-    ndim_supp = 2
+    ndim_supp = 1
     ndims_params = [1, 0, 2]
     dtype = "floatX"
     _print_name = ("KroneckerNormal", "\\operatorname{KroneckerNormal}")
 
-    def _shape_from_params(self, dist_params, rep_param_idx=0, param_shapes=None):
-        return default_shape_from_params(1, dist_params, rep_param_idx, param_shapes)
-
     def rng_fn(self, rng, mu, sigma, *covs, size=None):
         size = size if size else covs[-1]
         covs = covs[:-1] if covs[-1] == size else covs
@@ -1965,7 +1962,6 @@ def dist(cls, mu, covs=None, chols=None, evds=None, sigma=None, *args, **kwargs)
 
         mu = at.as_tensor_variable(mu)
 
-        # mean = median = mode = mu
         return super().dist([mu, sigma, *covs], **kwargs)
 
     def get_moment(rv, size, mu, covs, chols, evds):

pymc/tests/test_distributions_moments.py

@@ -1415,7 +1415,7 @@ def normal_sim(rng, mu, sigma, size):
         ),
     ],
 )
-def test_kronecker_normal_moments(mu, covs, size, expected):
+def test_kronecker_normal_moment(mu, covs, size, expected):
     with Model() as model:
         KroneckerNormal("x", mu=mu, covs=covs, size=size)
     assert_moment_is_expected(model, expected)