docs and changelog

CHANGELOG.md

@@ -2,6 +2,7 @@
 
 ## Features
 
+- Added functions for normal probability density function (`pybamm.normal_pdf`) and cumulative distribution function (`pybamm.normal_cdf`) ([#3999](https://github.com/pybamm-team/PyBaMM/pull/3999))
 - Updates multiprocess `Pool` in `BaseSolver.solve()` to be constructed with context `fork`. Adds small example for multiprocess inputs. ([#3974](https://github.com/pybamm-team/PyBaMM/pull/3974))
 - Added custom experiment steps ([#3835](https://github.com/pybamm-team/PyBaMM/pull/3835))
 - Added support for macOS arm64 (M-series) platforms. ([#3789](https://github.com/pybamm-team/PyBaMM/pull/3789))

docs/source/api/expression_tree/functions.rst

@@ -77,3 +77,7 @@ Functions
   :members:
 
 .. autofunction:: pybamm.tanh
+
+.. autofunction:: pybamm.normal_pdf
+
+.. autofunction:: pybamm.normal_cdf

pybamm/expression_tree/functions.py

@@ -657,11 +657,47 @@ def tanh(child: pybamm.Symbol):
     return simplified_function(Tanh, child)
 
 
-def normal_pdf(x, mu, sigma):
-    """Returns the normal probability distribution function at x."""
+def normal_pdf(
+    x: pybamm.Symbol, mu: pybamm.Symbol | float, sigma: pybamm.Symbol | float
+):
+    """
+    Returns the normal probability density function at x.
+
+    Parameters
+    ----------
+    x : pybamm.Symbol
+        The value at which to evaluate the normal distribution
+    mu : pybamm.Symbol or float
+        The mean of the normal distribution
+    sigma : pybamm.Symbol or float
+        The standard deviation of the normal distribution
+
+    Returns
+    -------
+    pybamm.Symbol
+        The value of the normal distribution at x
+    """
     return 1 / (np.sqrt(2 * np.pi) * sigma) * np.exp(-0.5 * ((x - mu) / sigma) ** 2)
 
 
-def normal_cdf(x, mu, sigma):
-    """Returns the normal cumulative distribution function at x."""
+def normal_cdf(
+    x: pybamm.Symbol, mu: pybamm.Symbol | float, sigma: pybamm.Symbol | float
+):
+    """
+    Returns the normal cumulative distribution function at x.
+
+    Parameters
+    ----------
+    x : pybamm.Symbol
+        The value at which to evaluate the normal distribution
+    mu : pybamm.Symbol or float
+        The mean of the normal distribution
+    sigma : pybamm.Symbol or float
+        The standard deviation of the normal distribution
+
+    Returns
+    -------
+    pybamm.Symbol
+        The value of the normal distribution at x
+    """
     return 0.5 * (1 + special.erf((x - mu) / (sigma * np.sqrt(2))))