Add a table method to Model

Author: ricardoV94

pymc/model/core.py

@@ -2036,6 +2036,26 @@ def to_graphviz(
             dpi=dpi,
         )
 
+    def table(
+        self,
+        *,
+        split_groups: bool = True,
+        truncate_deterministic: int | None = None,
+        parameter_count: bool = True,
+    ):
+        """Create a rich table summarizing the model's variables and their expressions.
+
+        See :func:`pymc.model_table` for details.
+        """
+        from pymc.printing import model_table
+
+        return model_table(
+            self,
+            split_groups=split_groups,
+            truncate_deterministic=truncate_deterministic,
+            parameter_count=parameter_count,
+        )
+
 
 class BlockModelAccess(Model):
     """Can be used to prevent user access to Model contexts."""

pymc/printing.py

@@ -420,7 +420,7 @@ def _extract_dim_value(var: Variable) -> np.ndarray:
         return " × ".join(f"{dim}[{dim_size}]" for dim, dim_size in dim_sizes.items())
     if not isinstance(var.type, HasShape):
         return ""
-    shape_values = list(pt.as_tensor(var.shape).eval(mode=_cheap_eval_mode))
+    shape_values = list(pt.as_tensor(var.shape).eval(mode=_cheap_eval_mode))  # type: ignore[attr-defined]
     return f"[{', '.join(map(str, shape_values))}]" if shape_values else ""