From 59be9b7a0450336f665becd0037baef6b588aa37 Mon Sep 17 00:00:00 2001 From: Peter Hawkins Date: Tue, 2 Jul 2019 15:00:47 -0400 Subject: [PATCH] Minor doc fixes. --- docs/jax.scipy.rst | 1 + jax/numpy/lax_numpy.py | 1 + 2 files changed, 2 insertions(+) diff --git a/docs/jax.scipy.rst b/docs/jax.scipy.rst index f054efb5f..d296e3230 100644 --- a/docs/jax.scipy.rst +++ b/docs/jax.scipy.rst @@ -85,6 +85,7 @@ jax.scipy.stats.laplace .. autosummary:: :toctree: _autosummary + cdf logpdf pdf diff --git a/jax/numpy/lax_numpy.py b/jax/numpy/lax_numpy.py index 3bdd0fc1c..67c12fb69 100644 --- a/jax/numpy/lax_numpy.py +++ b/jax/numpy/lax_numpy.py @@ -1247,6 +1247,7 @@ def _pad(array, pad_width, mode, constant_values): msg = "Unimplemented padding mode '{}' for np.pad." raise NotImplementedError(msg.format(mode)) +@_wraps(onp.pad) def pad(array, pad_width, mode, constant_values=0): return _pad(array, pad_width, mode, constant_values)