refactor: more selective filter_jit application

Signed-off-by: nstarman <nstarman@users.noreply.github.com>
GalacticDynamics · Sep 19, 2024 · b5f8e55 · b5f8e55
1 parent 5d2d55e
commit b5f8e55
Show file tree

Hide file tree

Showing 2 changed files with 0 additions and 4 deletions.
diff --git a/src/quaxed/array_api/_creation_functions.py b/src/quaxed/array_api/_creation_functions.py
@@ -22,7 +22,6 @@
 
 from typing import TypeVar
 
-import equinox as eqx
 import jax
 import jax.numpy as jnp
 from jax import Device
@@ -95,7 +94,6 @@ def arange(
 # =============================================================================
 
 
-@eqx.filter_jit  # type: ignore[misc]
 @quaxify
 def asarray(
     obj: ArrayLike,

diff --git a/src/quaxed/numpy/_creation_functions.py b/src/quaxed/numpy/_creation_functions.py
@@ -17,7 +17,6 @@
 
 from typing import Literal, TypeVar
 
-import equinox as eqx
 import jax
 import jax.numpy as jnp
 from jaxtyping import ArrayLike
@@ -85,7 +84,6 @@ def arange(
 # =============================================================================
 
 
-@eqx.filter_jit  # type: ignore[misc]
 @quaxify
 def asarray(
     obj: ArrayLike,