diff --git a/pymc/distributions/discrete.py b/pymc/distributions/discrete.py index d2c32c1a49e..6ed1a901da6 100644 --- a/pymc/distributions/discrete.py +++ b/pymc/distributions/discrete.py @@ -1745,7 +1745,7 @@ class OrderedLogistic: def __new__(cls, name, *args, compute_p=True, **kwargs): out_rv = _OrderedLogistic(name, *args, **kwargs) if compute_p: - pm.Deterministic(f"{name}_probs", out_rv.owner.inputs[3]) + pm.Deterministic(f"{name}_probs", out_rv.owner.inputs[3], dims=kwargs.get("dims")) return out_rv @classmethod @@ -1856,7 +1856,7 @@ class OrderedProbit: def __new__(cls, name, *args, compute_p=True, **kwargs): out_rv = _OrderedProbit(name, *args, **kwargs) if compute_p: - pm.Deterministic(f"{name}_probs", out_rv.owner.inputs[3]) + pm.Deterministic(f"{name}_probs", out_rv.owner.inputs[3], dims=kwargs.get("dims")) return out_rv @classmethod diff --git a/pymc/distributions/multivariate.py b/pymc/distributions/multivariate.py index ac25f7308cc..abeb4909994 100644 --- a/pymc/distributions/multivariate.py +++ b/pymc/distributions/multivariate.py @@ -752,7 +752,7 @@ class OrderedMultinomial: def __new__(cls, name, *args, compute_p=True, **kwargs): out_rv = _OrderedMultinomial(name, *args, **kwargs) if compute_p: - pm.Deterministic(f"{name}_probs", out_rv.owner.inputs[4]) + pm.Deterministic(f"{name}_probs", out_rv.owner.inputs[4], dims=kwargs.get("dims")) return out_rv @classmethod