Add dim name to families with categorical response levels

bambi/backend/terms.py

@@ -13,7 +13,7 @@
     GP_KERNELS,
 )
 from bambi.families.multivariate import MultivariateFamily, Multinomial, DirichletMultinomial
-from bambi.families.univariate import Categorical
+from bambi.families.univariate import Categorical, Cumulative, StoppingRatio
 from bambi.priors import Prior
 
 
@@ -234,7 +234,17 @@ def build(self, pymc_backend, bmb_model):
         # Auxiliary parameters and data
         kwargs = {"observed": data, "dims": ("__obs__",)}
 
-        if isinstance(self.family, (MultivariateFamily, Categorical)):
+        if isinstance(
+            self.family,
+            (
+                MultivariateFamily,
+                Categorical,
+                Cumulative,
+                StoppingRatio,
+                Multinomial,
+                DirichletMultinomial,
+            ),
+        ):
             response_term = bmb_model.response_component.term
             response_name = response_term.alias or response_term.name
             dim_name = response_name + "_dim"
@@ -386,10 +396,6 @@ def robustify_dims(self, pymc_backend, kwargs):
         # In this case, we add extra dimensions to avoid having shape mismatch between the data
         # and the shape implied by the `dims` we pass.
 
-        # Don't do it for the Multinomial families (it's an exception)
-        if isinstance(self.family, (Multinomial, DirichletMultinomial)):
-            return kwargs
-
         if (
             self.term.is_censored
             or self.term.is_truncated