From e3e01c8ef9f5193fd3958e636200cfe3b54ad0b8 Mon Sep 17 00:00:00 2001 From: Evgeni Burovski Date: Wed, 5 Feb 2025 10:44:32 +0100 Subject: [PATCH] MAINT: block axis=None in squeeze closes https://github.com/data-apis/array-api-strict/issues/62 --- array_api_strict/_manipulation_functions.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/array_api_strict/_manipulation_functions.py b/array_api_strict/_manipulation_functions.py index d775835..63c3516 100644 --- a/array_api_strict/_manipulation_functions.py +++ b/array_api_strict/_manipulation_functions.py @@ -153,6 +153,11 @@ def squeeze(x: Array, /, axis: Union[int, Tuple[int, ...]]) -> Array: See its docstring for more information. """ + if axis is None: + raise ValueError( + "squeeze(..., axis=None is not supported. See " + "https://github.com/data-apis/array-api/pull/100 for a discussion." + ) return Array._new(np.squeeze(x._array, axis=axis), device=x.device)