Skip to content

Commit 7ad3e27

Browse files
committed
fix: export Array
1 parent e4673de commit 7ad3e27

File tree

1 file changed

+3
-2
lines changed

1 file changed

+3
-2
lines changed

src/array_api_stubs/_draft/_types.py

+3-2
Original file line numberDiff line numberDiff line change
@@ -19,6 +19,7 @@
1919
"Union",
2020
"Sequence",
2121
"array",
22+
"Array",
2223
"device",
2324
"dtype",
2425
"ellipsis",
@@ -46,7 +47,7 @@
4647
from enum import Enum
4748
from .data_types import DType
4849

49-
array = TypeVar("array", bound="_array")
50+
array = TypeVar("array", bound="Array")
5051
device = TypeVar("device")
5152
dtype = TypeVar("dtype", bound=DType)
5253
device_ = TypeVar("device_") # only used in this file
@@ -149,7 +150,7 @@ def dtypes(
149150
)
150151

151152

152-
class _array(Protocol[array, dtype_, device_, PyCapsule]): # type: ignore
153+
class Array(Protocol[array, dtype_, device_, PyCapsule]): # type: ignore
153154
def __init__(self: array) -> None:
154155
"""Initialize the attributes for the array object class."""
155156

0 commit comments

Comments
 (0)