fix docstrings

scikit-hep · Aug 14, 2023 · b48e544 · b48e544
1 parent a4c1acb
commit b48e544
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 9 deletions.
diff --git a/src/pyhf/tensor/jax_backend.py b/src/pyhf/tensor/jax_backend.py
@@ -648,10 +648,10 @@ def diagonal(self, tensor_in):
             >>> pyhf.set_backend("jax")
             >>> tensor = pyhf.tensorlib.astensor([[1.0, 0.0], [0.0, 1.0]])
             >>> tensor
-            tensor([[1., 0.],
-                    [0., 1.]])
+            Array([[1., 0.],
+                   [0., 1.]], dtype=float64)
             >>> pyhf.tensorlib.diagonal(tensor)
-            tensor([1., 1.])
+            Array([1., 1.], dtype=float64)
 
         Args:
             tensor_in (:obj:`tensor`): The input tensor object.

diff --git a/src/pyhf/tensor/numpy_backend.py b/src/pyhf/tensor/numpy_backend.py
@@ -668,10 +668,10 @@ def diagonal(self, tensor_in: Tensor[T]) -> ArrayLike:
             >>> pyhf.set_backend("numpy")
             >>> tensor = pyhf.tensorlib.astensor([[1.0, 0.0], [0.0, 1.0]])
             >>> tensor
-            tensor([[1., 0.],
-                    [0., 1.]])
+            array([[1., 0.],
+                   [0., 1.]])
             >>> pyhf.tensorlib.diagonal(tensor)
-            tensor([1., 1.])
+            array([1., 1.])
 
         Args:
             tensor_in (:obj:`tensor`): The input tensor object.

diff --git a/src/pyhf/tensor/tensorflow_backend.py b/src/pyhf/tensor/tensorflow_backend.py
@@ -752,10 +752,11 @@ def diagonal(self, tensor_in):
             >>> pyhf.set_backend("tensorflow")
             >>> tensor = pyhf.tensorlib.astensor([[1.0, 0.0], [0.0, 1.0]])
             >>> tensor
-            tensor([[1., 0.],
-                    [0., 1.]])
+            <tf.Tensor: shape=(2, 2), dtype=float64, numpy=
+            array([[1., 0.],
+                   [0., 1.]])>
             >>> pyhf.tensorlib.diagonal(tensor)
-            tensor([1., 1.])
+            <tf.Tensor: shape=(2), dtype=float64, numpy=array([1., 1.])
 
         Args:
             tensor_in (:obj:`tensor`): The input tensor object.