From 17740374cd72def41386e76ff4b3be3f9b36a49e Mon Sep 17 00:00:00 2001 From: crusaderky Date: Fri, 14 Mar 2025 09:40:13 +0000 Subject: [PATCH] MAINT: Allow `import xpx.testing` outside of test modules --- src/array_api_extra/testing.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) 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