From bed80fbba83a16f936aebadfb242f63a85e9f1aa Mon Sep 17 00:00:00 2001 From: Evgeni Burovski Date: Thu, 19 Feb 2026 15:30:35 +0100 Subject: [PATCH] API: expand_dims axis argument kwd or positional In the spec, `expand_dims` accepts axis as either keyword or positional argument, cf https://data-apis.org/array-api/2024.12/API_specification/generated/array_api.expand_dims.html#array_api.expand_dims --- array_api_strict/_manipulation_functions.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/array_api_strict/_manipulation_functions.py b/array_api_strict/_manipulation_functions.py index a04b18e..c2dcccf 100644 --- a/array_api_strict/_manipulation_functions.py +++ b/array_api_strict/_manipulation_functions.py @@ -30,7 +30,7 @@ def concat( ) -def expand_dims(x: Array, /, *, axis: int) -> Array: +def expand_dims(x: Array, /, axis: int) -> Array: """ Array API compatible wrapper for :py:func:`np.expand_dims `.