Skip to content

Commit cda6110

Browse files
matthiasdienerinducer
authored andcommitted
EagerJAXArrayContext: remove stray-ish einsum warning
1 parent 0abca20 commit cda6110

File tree

1 file changed

+0
-5
lines changed

1 file changed

+0
-5
lines changed

arraycontext/impl/jax/__init__.py

Lines changed: 0 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -132,11 +132,6 @@ def call_loopy(self, t_unit, **kwargs):
132132

133133
def einsum(self, spec, *args, arg_names=None, tagged=()):
134134
import jax.numpy as jnp
135-
if arg_names is not None:
136-
from warnings import warn
137-
warn("'arg_names' don't bear any significance in "
138-
f"{type(self).__name__}.", stacklevel=2)
139-
140135
return jnp.einsum(spec, *args)
141136

142137
def clone(self):

0 commit comments

Comments
 (0)