Remove Untyped

data-apis · Dec 12, 2024 · 8a686f8 · 8a686f8
1 parent d3a5bc4
commit 8a686f8
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 7 deletions.
diff --git a/src/array_api_extra/_lib/_typing.py b/src/array_api_extra/_lib/_typing.py
@@ -2,9 +2,8 @@
 from typing import Any
 
 # To be changed to a Protocol later (see data-apis/array-api#589)
-Untyped = Any  # type: ignore[no-any-explicit]
-Array = Untyped
-Device = Untyped
-Index = Untyped
+Array = Any  # type: ignore[no-any-explicit]
+Device = Any  # type: ignore[no-any-explicit]
+Index = Any  # type: ignore[no-any-explicit]
 
-__all__ = ["Array", "Device", "Index", "ModuleType", "Untyped"]
+__all__ = ["Array", "Device", "Index", "ModuleType"]
diff --git a/tests/test_at.py b/tests/test_at.py
@@ -17,7 +17,7 @@
 from array_api_extra import at
 
 if TYPE_CHECKING:
-    from array_api_extra._lib._typing import Array, Untyped
+    from array_api_extra._lib._typing import Array
 
 all_libraries = (
     "array_api_strict",
@@ -95,7 +95,7 @@ def assert_copy(array: Array, copy: bool | None) -> Generator[None, None, None]:
 )
 def test_update_ops(
     array: Array,
-    kwargs: dict[str, Untyped],
+    kwargs: dict[str, bool | None],
     expect_copy: bool | None,
     op: str,
     arg: float,