fix stopping ratio family

bambinos · Apr 9, 2024 · b431d81 · b431d81
1 parent c927f64
commit b431d81
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/bambi/families/univariate.py b/bambi/families/univariate.py
@@ -403,7 +403,7 @@ def transform_backend_eta(eta, kwargs):
     def transform_backend_kwargs(kwargs):
         # P(Y = k) = F(threshold_k - eta) * \prod_{j=1}^{k-1}{1 - F(threshold_j - eta)}
         p = kwargs.pop("p")
-        n_columns = p.type.shape[-1]
+        n_columns = p.shape.eval()[-1]
         p = pt.concatenate(
             [
                 pt.shape_padright(p[..., 0]),