Merge pull request #240 from jpreszler/issue_238_math_rendering

improve math rendering of pymc model docs
pymc-labs · Sep 18, 2023 · 28d50b9 · 28d50b9
2 parents 234a0cd + e2511f9
commit 28d50b9
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 9 deletions.
diff --git a/causalpy/pymc_models.py b/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
     --------

diff --git a/pyproject.toml b/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",