diff --git a/src/array_api_extra/testing.py b/src/array_api_extra/testing.py index 550665ae..8aca160f 100644 --- a/src/array_api_extra/testing.py +++ b/src/array_api_extra/testing.py @@ -13,8 +13,6 @@ from types import ModuleType from typing import TYPE_CHECKING, Any, TypeVar, cast -import pytest - from ._lib._utils._compat import is_dask_namespace, is_jax_namespace __all__ = ["lazy_xp_function", "patch_lazy_xp_functions"] @@ -25,6 +23,7 @@ # https://github.com/scikit-learn/scikit-learn/pull/27910#issuecomment-2568023972 from typing import ParamSpec + import pytest from dask.typing import Graph, Key, SchedulerGetCallable from typing_extensions import override