Skip to content

Commit

Permalink
docs: Update JAX backend array_type to jaxlib.xla_extension.ArrayImpl (
Browse files Browse the repository at this point in the history
…#2134)

* Update JAX backend array_type to jaxlib.xla_extension.ArrayImpl, which is a unified
  array type introduced in jaxlib v0.4.6.
   - c.f. https://github.com/google/jax/releases/tag/jaxlib-v0.4.6
   - c.f. jax-ml/jax#14768
  • Loading branch information
matthewfeickert authored Mar 13, 2023
1 parent a4ec176 commit 48e7f28
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions src/pyhf/tensor/jax_backend.py
Original file line number Diff line number Diff line change
Expand Up @@ -216,13 +216,13 @@ def astensor(self, tensor_in, dtype="float"):
Array([[1., 2., 3.],
[4., 5., 6.]], dtype=float64)
>>> type(tensor) # doctest:+ELLIPSIS
<class '...Array'>
<class '...ArrayImpl'>
Args:
tensor_in (Number or Tensor): Tensor object
Returns:
`jaxlib.xla_extension.Array`: A multi-dimensional, fixed-size homogeneous array.
`jaxlib.xla_extension.ArrayImpl`: A multi-dimensional, fixed-size homogeneous array.
"""
# TODO: Remove doctest:+ELLIPSIS when JAX API stabilized
try:
Expand Down

0 comments on commit 48e7f28

Please sign in to comment.