Skip to content

Commit 53e1663

Browse files
committed
Fix reference to jax array type
1 parent 50511fe commit 53e1663

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

arraycontext/impl/jax/__init__.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -41,7 +41,7 @@
4141
class EagerJAXArrayContext(ArrayContext):
4242
"""
4343
A :class:`ArrayContext` that uses
44-
:class:`jaxlib.xla_extension.DeviceArrayBase` instances for its base array
44+
:class:`jax.Array` instances for its base array
4545
class and performs all array operations eagerly. See
4646
:class:`~arraycontext.PytatoJAXArrayContext` for a lazier version.
4747

0 commit comments

Comments
 (0)