implement _display_ method for model

Author: williambdean

pymc/model/core.py

@@ -52,7 +52,7 @@
 from pymc.logprob.basic import transformed_conditional_logp
 from pymc.logprob.transforms import Transform
 from pymc.logprob.utils import ParameterValueError, replace_rvs_by_values
-from pymc.model_graph import model_to_graphviz
+from pymc.model_graph import model_to_graphviz, model_to_mermaid
 from pymc.pytensorf import (
     PointFunc,
     SeedSequenceSeed,
@@ -437,6 +437,11 @@ def __exit__(self, exc_type: None, exc_val: None, exc_tb: None) -> None:
         """Exit the context manager."""
         _ = MODEL_MANAGER.active_contexts.pop()
 
+    def _display_(self):
+        import marimo as mo
+
+        return mo.mermaid(model_to_mermaid(self))
+
     @staticmethod
     def _validate_name(name):
         if name.endswith(":"):