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`.
scikit-hep · Mar 17, 2022 · 1884c6c · 1884c6c
1 parent f862b8c
commit 1884c6c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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])