Adapt to draft_version

Author: betatim

array_api_strict/_searching_functions.py

@@ -89,7 +89,7 @@ def where(condition: Array, x1: bool | int | float | Array, x2: bool | int | flo
     _result_type(x1.dtype, x2.dtype)
 
     if len({a.device for a in (condition, x1, x2)}) > 1:
-        raise ValueError("where inputs must all be on the same device")
+        raise ValueError("Inputs to `where` must all use the same device")
 
     x1, x2 = Array._normalize_two_args(x1, x2)
     return Array._new(np.where(condition._array, x1._array, x2._array), device=x1.device)

array_api_strict/tests/test_searching_functions.py

@@ -3,7 +3,7 @@
 import array_api_strict as xp
 
 from array_api_strict import ArrayAPIStrictFlags
-from array_api_strict._flags import next_supported_version
+from array_api_strict._flags import draft_version
 
 
 def test_where_with_scalars():
@@ -14,7 +14,7 @@ def test_where_with_scalars():
         xp.where(x == 1, 42, 44)
 
     # Versions after 2023.12 support scalar arguments
-    with ArrayAPIStrictFlags(api_version=next_supported_version):
+    with ArrayAPIStrictFlags(api_version=draft_version):
         x_where = xp.where(x == 1, 42, 44)
 
         expected = xp.asarray([42, 44, 44, 42])