Merge pull request #240 from jpreszler/issue_238_math_rendering

improve math rendering of pymc model docs

Author: drbenvincent

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
     --------

pyproject.toml

@@ -31,7 +31,7 @@ dependencies = [
     "graphviz",
     "ipython!=8.7.0",
     "matplotlib>=3.5.3",
-    "numpy",
+    "numpy<1.26.0",
     "pandas",
     "patsy",
     "pymc>=5.0.0",