diff --git a/src/spexial/_src/spence.py b/src/spexial/_src/spence.py index e0896a2..916a72f 100644 --- a/src/spexial/_src/spence.py +++ b/src/spexial/_src/spence.py @@ -13,6 +13,7 @@ MAXITER = 500 +@jax.numpy.vectorize def complex_spence_series_0(z): """ Small z branch, uses a series expansion about :math:`z = 0` @@ -27,6 +28,7 @@ def complex_spence_series_0(z): return np.pi**2 / 6 - sum1 + jnp.log(z) * sum2 +@jax.numpy.vectorize def complex_spence_series_1(z): """ Middle branch, an expansion around :math:`z = 1`