fix: export array

data-apis · Nov 24, 2024 · ac20ed5 · ac20ed5
1 parent e4673de
commit ac20ed5
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 5 deletions.
diff --git a/spec/draft/API_specification/array_object.rst b/spec/draft/API_specification/array_object.rst
@@ -275,7 +275,7 @@ Methods
    Array.__abs__
    Array.__add__
    Array.__and__
-   Array.__Array_namespace__
+   Array.__array_namespace__
    Array.__bool__
    Array.__complex__
    Array.__dlpack__

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."""
 

diff --git a/src/array_api_stubs/_draft/array_object.py b/src/array_api_stubs/_draft/array_object.py
@@ -1,3 +1,3 @@
-from ._types import array
+from ._types import array, Array
 
-__all__ = ["array"]
+__all__ = ["array", "Array"]