Skip to content

Commit 956ed6e

Browse files
fix: Update JAX DeviceArray type in docstrings for doctest
* Update JAX DeviceArray type in docstrings to jax.interpreters.xla._DeviceArray - Needed to pass doctest
1 parent 8a06262 commit 956ed6e

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

src/pyhf/tensor/jax_backend.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -210,13 +210,13 @@ def astensor(self, tensor_in, dtype='float'):
210210
DeviceArray([[1., 2., 3.],
211211
[4., 5., 6.]], dtype=float64)
212212
>>> type(tensor)
213-
<class 'jax.interpreters.xla.DeviceArray'>
213+
<class 'jax.interpreters.xla._DeviceArray'>
214214
215215
Args:
216216
tensor_in (Number or Tensor): Tensor object
217217
218218
Returns:
219-
`jax.interpreters.xla.DeviceArray`: A multi-dimensional, fixed-size homogenous array.
219+
`jax.interpreters.xla._DeviceArray`: A multi-dimensional, fixed-size homogenous array.
220220
"""
221221
try:
222222
dtype = self.dtypemap[dtype]

0 commit comments

Comments
 (0)