implemented fix for escaping underscores in latex repr and added a un… (pymc-devs#7501)

* implemented fix for escaping underscores in latex repr and added a unit test

* updated unit test staticmethod to include underscore in var name

* add underscore escape fix to distribution repr as well as model repr, fixed testing to expect underscores in LaTeX representation to be escaped

* added cleaner method using re to escape underscores, added cleaner test to assert underscores are escaped

Author: Dekermanjian

pymc/printing.py

@@ -37,11 +37,7 @@
 def str_for_dist(
     dist: TensorVariable, formatting: str = "plain", include_params: bool = True
 ) -> str:
-    """Make a human-readable string representation of a Distribution in a model.
-
-    This can be either LaTeX or plain, optionally with distribution parameter
-    values included.
-    """
+    """Make a human-readable string representation of a Distribution in a model, either LaTeX or plain, optionally with distribution parameter values included."""
     if include_params:
         if isinstance(dist.owner.op, RandomVariable) or getattr(
             dist.owner.op, "extended_signature", None
@@ -100,10 +96,8 @@ def str_for_dist(
 
 
 def str_for_model(model: Model, formatting: str = "plain", include_params: bool = True) -> str:
-    """Make a human-readable string representation of Model.
-
-    This lists all random variables and their distributions, optionally
-    including parameter values.
+    """Make a human-readable string representation of Model, listing all random variables
+    and their distributions, optionally including parameter values.
     """
     # Wrap functions to avoid confusing typecheckers
     sfd = partial(str_for_dist, formatting=formatting, include_params=include_params)
@@ -150,10 +144,8 @@ def str_for_potential_or_deterministic(
     include_params: bool = True,
     dist_name: str = "Deterministic",
 ) -> str:
-    """Make a human-readable string representation of a Deterministic or Potential in a model.
-
-    This can be either LaTeX or plain, optionally with distribution parameter
-    values included.
+    """Make a human-readable string representation of a Deterministic or Potential in a model, either
+    LaTeX or plain, optionally with distribution parameter values included.
     """
     print_name = var.name if var.name is not None else "<unnamed>"
     if "latex" in formatting:
@@ -309,5 +301,7 @@ def _default_repr_pretty(obj: TensorVariable | Model, p, cycle):
 
 
 def _format_underscore(variable: str) -> str:
-    """Escapes all unescaped underscores in the variable name for LaTeX representation."""
+    """
+    Escapes all unescaped underscores in the variable name for LaTeX representation.
+    """
     return re.sub(r"(?<!\\)_", r"\\_", variable)