improve math rendering of pymc model docs

Author: jpreszler

causalpy/pymc_models.py

@@ -152,13 +152,13 @@ class WeightedSumFitter(ModelBuilder):
 
     .. math::
 
-        sigma \sim HalfNormal(1)
+        \sigma &\sim \mathrm{HalfNormal}(1)
 
-        beta \sim Dirichlet(1,...,1)
+        \\beta &\sim \mathrm{Dirichlet}(1,...,1)
 
-        mu = X * beta
+        \mu &= X * \\beta
 
-        y \sim Normal(mu, sigma)
+        y &\sim \mathrm{Normal}(\mu, \sigma)
 
     Example
     --------
@@ -204,13 +204,13 @@ class LinearRegression(ModelBuilder):
     Defines the PyMC model
 
     .. math::
-        beta \sim Normal(0, 50)
+        \\beta &\sim \mathrm{Normal}(0, 50)
 
-        sigma \sim HalfNormal(1)
+        \sigma &\sim \mathrm{HalfNormal}(1)
 
-        mu = X * beta
+        \mu &= X * \\beta
 
-        y \sim Normal(mu, sigma)
+        y &\sim \mathrm{Normal}(\mu, \sigma)
 
     Example
     --------