Skip to content

Commit 6d59ae8

Browse files
committed
Fix a comment
1 parent 9c8bed6 commit 6d59ae8

File tree

1 file changed

+2
-3
lines changed

1 file changed

+2
-3
lines changed

array_api_compat/common/_helpers.py

Lines changed: 2 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -125,9 +125,8 @@ def your_function(x, y):
125125
raise TypeError("_use_compat cannot be False if input array is a dask array!")
126126
elif is_jax_array(x):
127127
_check_api_version(api_version)
128-
# jax.numpy is already an array namespace, but requires this
129-
# side-effecting import for __array_namespace__ and some other
130-
# things to be defined.
128+
# jax.experimental.array_api is already an array namespace. We do
129+
# not have a wrapper submodule for it.
131130
import jax.experimental.array_api as jnp
132131
namespaces.add(jnp)
133132
elif hasattr(x, '__array_namespace__'):

0 commit comments

Comments
 (0)