Merge pull request data-apis#76 from lucascolley/annotations

TYP: import annotations for sklearn
lucascolley · Jan 3, 2025 · 346365d · 346365d
2 parents 84bf725 + b6d43d1
commit 346365d
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 1 deletion.
diff --git a/src/array_api_extra/_funcs.py b/src/array_api_extra/_funcs.py
@@ -1,5 +1,8 @@
 """Public API Functions."""
 
+# https://github.com/scikit-learn/scikit-learn/pull/27910#issuecomment-2568023972
+from __future__ import annotations
+
 import operator
 import warnings
 
@@ -719,7 +722,7 @@ def __init__(
         self._x = x
         self._idx = idx
 
-    def __getitem__(self, idx: Index, /) -> "at":  # numpydoc ignore=PR01,RT01
+    def __getitem__(self, idx: Index, /) -> at:  # numpydoc ignore=PR01,RT01
         """
         Allow for the alternate syntax ``at(x)[start:stop:step]``.
 

diff --git a/src/array_api_extra/_lib/_compat.pyi b/src/array_api_extra/_lib/_compat.pyi
@@ -1,5 +1,8 @@
 """Static type stubs for `_compat.py`."""
 
+# https://github.com/scikit-learn/scikit-learn/pull/27910#issuecomment-2568023972
+from __future__ import annotations
+
 from types import ModuleType
 
 from ._typing import Array, Device

diff --git a/src/array_api_extra/_lib/_utils.py b/src/array_api_extra/_lib/_utils.py
@@ -1,5 +1,8 @@
 """Utility functions used by `array_api_extra/_funcs.py`."""
 
+# https://github.com/scikit-learn/scikit-learn/pull/27910#issuecomment-2568023972
+from __future__ import annotations
+
 from . import _compat
 from ._typing import Array, ModuleType