From 53e16634ffab956203f741b357376ff1b6790f0a Mon Sep 17 00:00:00 2001
From: Andreas Kloeckner <inform@tiker.net>
Date: Sun, 19 Feb 2023 15:46:22 -0600
Subject: [PATCH] Fix reference to jax array type
---
arraycontext/impl/jax/__init__.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/arraycontext/impl/jax/__init__.py b/arraycontext/impl/jax/__init__.py
index e680f7e..e5fef3e 100644
--- a/arraycontext/impl/jax/__init__.py
+++ b/arraycontext/impl/jax/__init__.py
@@ -41,7 +41,7 @@ from arraycontext.context import Array, ArrayContext, ArrayOrContainer, ScalarLi
class EagerJAXArrayContext(ArrayContext):
"""
A :class:`ArrayContext` that uses
- :class:`jaxlib.xla_extension.DeviceArrayBase` instances for its base array
+ :class:`jax.Array` instances for its base array
class and performs all array operations eagerly. See
:class:`~arraycontext.PytatoJAXArrayContext` for a lazier version.
--
GitLab