From b5f8e557bfb849420834f5ce48579f4dfc240cf2 Mon Sep 17 00:00:00 2001 From: nstarman Date: Wed, 18 Sep 2024 23:43:36 -0400 Subject: [PATCH] refactor: more selective filter_jit application Signed-off-by: nstarman --- src/quaxed/array_api/_creation_functions.py | 2 -- src/quaxed/numpy/_creation_functions.py | 2 -- 2 files changed, 4 deletions(-) diff --git a/src/quaxed/array_api/_creation_functions.py b/src/quaxed/array_api/_creation_functions.py index b177cf5..c03a824 100644 --- 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 index 1d219b2..bf2f1f9 100644 --- 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,