From 8a686f83cf3f3287fb981192a65d740394c58a8f Mon Sep 17 00:00:00 2001
From: crusaderky <crusaderky@gmail.com>
Date: Thu, 12 Dec 2024 14:56:10 +0000
Subject: [PATCH] Remove Untyped

---
 src/array_api_extra/_lib/_typing.py | 9 ++++-----
 tests/test_at.py                    | 4 ++--
 2 files changed, 6 insertions(+), 7 deletions(-)

diff --git a/src/array_api_extra/_lib/_typing.py b/src/array_api_extra/_lib/_typing.py
index 101afd9..12e4431 100644
--- 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
index 2b43cb9..62e22aa 100644
--- 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,