Remove from __future__ import annotations

src/array_api_extra/_funcs.py

@@ -1,5 +1,3 @@
-from __future__ import annotations
-
 import operator
 import warnings
 
@@ -672,7 +670,7 @@ def __init__(self, x: Array, idx: Index = _undef, /) -> None:
         self._x = x
         self._idx = idx
 
-    def __getitem__(self, idx: Index, /) -> at:
+    def __getitem__(self, idx: Index, /) -> "at":
         """Allow for the alternate syntax ``at(x)[start:stop:step]``,
         which looks prettier than ``at(x, slice(start, stop, step))``
         and feels more intuitive coming from the JAX documentation.

tests/test_at.py

@@ -1,9 +1,6 @@
-from __future__ import annotations
-
 from collections.abc import Generator
 from contextlib import contextmanager
 from importlib import import_module
-from typing import TYPE_CHECKING
 
 import numpy as np
 import pytest
@@ -15,9 +12,7 @@
 )
 
 from array_api_extra import at
-
-if TYPE_CHECKING:
-    from array_api_extra._lib._typing import Array
+from array_api_extra._lib._typing import Array
 
 all_libraries = (
     "array_api_strict",