From 1884c6c08bc5b629724dd9ef3c60ec4386353b94 Mon Sep 17 00:00:00 2001 From: Matthew Feickert Date: Thu, 17 Mar 2022 00:04:56 -0500 Subject: [PATCH] docs: Update JAX backend normal docstring to jax v0.3.2 returns (#1813) * Update jax backend normal docstring examples to jax v0.3.2+ DeviceArray return of showing `weak_type=True`. --- src/pyhf/tensor/jax_backend.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/src/pyhf/tensor/jax_backend.py b/src/pyhf/tensor/jax_backend.py index db3b6dcca9..5e4a65bc80 100644 --- a/src/pyhf/tensor/jax_backend.py +++ b/src/pyhf/tensor/jax_backend.py @@ -477,7 +477,7 @@ def normal(self, x, mu, sigma): >>> import pyhf >>> pyhf.set_backend("jax") >>> pyhf.tensorlib.normal(0.5, 0., 1.) - DeviceArray(0.35206533, dtype=float64) + DeviceArray(0.35206533, dtype=float64, weak_type=True) >>> values = pyhf.tensorlib.astensor([0.5, 2.0]) >>> means = pyhf.tensorlib.astensor([0., 2.3]) >>> sigmas = pyhf.tensorlib.astensor([1., 0.8])