FIX Fix array API tests for array_api_strict 2.0.1 (#29387)

OmarManzoor · web-flow · commit 012de1e9fcd2 · 2024-07-03T17:04:49.000+02:00
diff --git a/sklearn/utils/_array_api.py b/sklearn/utils/_array_api.py
@@ -557,6 +557,11 @@ def get_namespace(*arrays, remove_none=True, remove_types=(str,), xp=None):
     if namespace.__name__ in {"cupy.array_api"}:
         namespace = _ArrayAPIWrapper(namespace)
 
+    if namespace.__name__ == "array_api_strict" and hasattr(
+        namespace, "set_array_api_strict_flags"
+    ):
+        namespace.set_array_api_strict_flags(api_version="2023.12")
+
     return namespace, is_array_api_compliant
 
 
