Skip to content

Commit f6caaa8

Browse files
committed
Increase __array_api_version__ to 2023.12
Also changes docstrings in _array_api.py
1 parent b1016bf commit f6caaa8

File tree

1 file changed

+15
-4
lines changed

1 file changed

+15
-4
lines changed

dpctl/tensor/_array_api.py

Lines changed: 15 additions & 4 deletions
Original file line numberDiff line numberDiff line change
@@ -49,7 +49,7 @@ def _isdtype_impl(dtype, kind):
4949
raise TypeError(f"Unsupported data type kind: {kind}")
5050

5151

52-
__array_api_version__ = "2022.12"
52+
__array_api_version__ = "2023.12"
5353

5454

5555
class Info:
@@ -80,6 +80,8 @@ def __init__(self):
8080

8181
def capabilities(self):
8282
"""
83+
capabilities()
84+
8385
Returns a dictionary of `dpctl`'s capabilities.
8486
8587
Returns:
@@ -92,12 +94,16 @@ def capabilities(self):
9294

9395
def default_device(self):
9496
"""
97+
default_device()
98+
9599
Returns the default SYCL device.
96100
"""
97101
return dpctl.select_default_device()
98102

99-
def default_dtypes(self, device=None):
103+
def default_dtypes(self, *, device=None):
100104
"""
105+
default_dtypes(*, device=None)
106+
101107
Returns a dictionary of default data types for `device`.
102108
103109
Args:
@@ -129,8 +135,10 @@ def default_dtypes(self, device=None):
129135
"indexing": dpt.dtype(default_device_index_type(device)),
130136
}
131137

132-
def dtypes(self, device=None, kind=None):
138+
def dtypes(self, *, device=None, kind=None):
133139
"""
140+
dtypes(*, device=None, kind=None)
141+
134142
Returns a dictionary of all Array API data types of a specified `kind`
135143
supported by `device`
136144
@@ -193,13 +201,16 @@ def dtypes(self, device=None, kind=None):
193201

194202
def devices(self):
195203
"""
204+
devices()
205+
196206
Returns a list of supported devices.
197207
"""
198208
return dpctl.get_devices()
199209

200210

201211
def __array_namespace_info__():
202-
"""__array_namespace_info__()
212+
"""
213+
__array_namespace_info__()
203214
204215
Returns a namespace with Array API namespace inspection utilities.
205216

0 commit comments

Comments
 (0)