Merge branch 'main' into exclude-dist-from-interrogate

Author: drbenvincent

README.md

@@ -198,6 +198,6 @@ Here are some general resources about causal inference:
 
 <img src="docs/source/_static/pymc-labs-log.png" align="right" width="50%" />
 
-This repository is supported by [PyMC Labs](https://www.pymc-labs.io).
+This repository is supported by [PyMC Labs](https://www.pymc-labs.com).
 
-If you are interested in seeing what PyMC Labs can do for you, then please email [ben.vincent@pymc-labs.io](mailto:ben.vincent@pymc-labs.io). We work with companies at a variety of scales and with varying levels of existing modeling capacity. We also run corporate workshop training events and can provide sessions ranging from introduction to Bayes to more advanced topics.
+If you are interested in seeing what PyMC Labs can do for you, then please email [ben.vincent@pymc-labs.com](mailto:ben.vincent@pymc-labs.com). We work with companies at a variety of scales and with varying levels of existing modeling capacity. We also run corporate workshop training events and can provide sessions ranging from introduction to Bayes to more advanced topics.

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