File tree Expand file tree Collapse file tree
Expand file tree Collapse file tree Original file line number Diff line number Diff line change @@ -251,7 +251,7 @@ def patch_lazy_xp_functions(
251251 """
252252 Test lazy execution of functions tagged with :func:`lazy_xp_function`.
253253
254- If ``xp==jax.numpy``, search for all functions and classes which have been tagged
254+ If ``xp==jax.numpy``, search for all functions and methods which have been tagged
255255 with :func:`lazy_xp_function` in the globals of the module that defines the current
256256 test, as well as in the ``lazy_xp_modules`` list in the globals of the same module,
257257 and wrap them with :func:`jax.jit`.
@@ -311,7 +311,7 @@ def xp(request):
311311 for target in search_targets :
312312 for obj_name in dir (target ):
313313 obj = getattr (target , obj_name )
314- if isinstance (obj , type ) and isinstance ( obj , Exception ) :
314+ if isinstance (obj , type ):
315315 classes .add (obj )
316316 search_targets .extend (classes )
317317
You can’t perform that action at this time.
0 commit comments