Fix optional keyword argument in take signature (#644)

This commit fixes the function signature for `take`. Namely, when
an input array is one-dimensional, the `axis` kwarg is optional;
when the array has more than one dimension, the `axis` kwarg is
required. Unfortunately, the type signature cannot encode this
duality, and we must rely on the specification text to clarify
that the `axis` kwarg is required for arrays having ranks greater
than unity.

Ref: data-apis/array-api-compat#34

Author: kgryte

src/array_api_stubs/_2022_12/indexing_functions.py

@@ -1,7 +1,7 @@
-from ._types import Union, array
+from ._types import Union, Optional, array
 
 
-def take(x: array, indices: array, /, *, axis: int) -> array:
+def take(x: array, indices: array, /, *, axis: Optional[int] = None) -> array:
     """
     Returns elements of an array along an axis.
 

src/array_api_stubs/_draft/indexing_functions.py

@@ -1,7 +1,7 @@
-from ._types import Union, array
+from ._types import Union, Optional, array
 
 
-def take(x: array, indices: array, /, *, axis: int) -> array:
+def take(x: array, indices: array, /, *, axis: Optional[int] = None) -> array:
     """
     Returns elements of an array along an axis.