Add max dimensions to the Capabilities typing dict

asmeurer · asmeurer · commit 632e895af7b9 · 2024-11-04T14:38:28.000-07:00
I don't know how to make this depend on API version so for now it's just there
always.
diff --git a/array_api_strict/_typing.py b/array_api_strict/_typing.py
@@ -54,7 +54,8 @@ class SupportsDLPack(Protocol):
     def __dlpack__(self, /, *, stream: None = ...) -> PyCapsule: ...
 
 Capabilities = TypedDict(
-    "Capabilities", {"boolean indexing": bool, "data-dependent shapes": bool}
+    "Capabilities", {"boolean indexing": bool, "data-dependent shapes": bool,
+                     "max dimensions": int}
 )
 
 DefaultDataTypes = TypedDict(

Original file line number	Diff line number	Diff line change
`@@ -54,7 +54,8 @@ class SupportsDLPack(Protocol):`
`54`	`54`	`def __dlpack__(self, /, *, stream: None = ...) -> PyCapsule: ...`
`55`	`55`
`56`	`56`	`Capabilities = TypedDict(`
`57`		`- "Capabilities", {"boolean indexing": bool, "data-dependent shapes": bool}`
	`57`	`+ "Capabilities", {"boolean indexing": bool, "data-dependent shapes": bool,`
	`58`	`+ "max dimensions": int}`
`58`	`59`	`)`
`59`	`60`
`60`	`61`	`DefaultDataTypes = TypedDict(`