fix: export Array

data-apis · Nov 24, 2024 · 7ad3e27 · 7ad3e27
1 parent e4673de
commit 7ad3e27
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/src/array_api_stubs/_draft/_types.py b/src/array_api_stubs/_draft/_types.py
@@ -19,6 +19,7 @@
     "Union",
     "Sequence",
     "array",
+    "Array",
     "device",
     "dtype",
     "ellipsis",
@@ -46,7 +47,7 @@
 from enum import Enum
 from .data_types import DType
 
-array = TypeVar("array", bound="_array")
+array = TypeVar("array", bound="Array")
 device = TypeVar("device")
 dtype = TypeVar("dtype", bound=DType)
 device_ = TypeVar("device_")  # only used in this file
@@ -149,7 +150,7 @@ def dtypes(
 )
 
 
-class _array(Protocol[array, dtype_, device_, PyCapsule]):  # type: ignore
+class Array(Protocol[array, dtype_, device_, PyCapsule]):  # type: ignore
     def __init__(self: array) -> None:
         """Initialize the attributes for the array object class."""