2024.12 scalars

Author: crusaderky

src/array_api_extra/_lib/_funcs.py

@@ -161,8 +161,7 @@ def _apply_where(  # type: ignore[no-any-explicit]  # numpydoc ignore=PR01,RT01
     temp1 = f1(*(arr[cond] for arr in args))
 
     if f2 is None:
-        # TODO remove asarrays once all backends support Array API 2024.12
-        dtype = xp.result_type(*asarrays(temp1, fill_value, xp=xp))
+        dtype = xp.result_type(temp1, fill_value)
         if getattr(fill_value, "ndim", 0):
             out = xp.astype(fill_value, dtype, copy=True)
         else:

tests/test_funcs.py

@@ -256,8 +256,11 @@ def f2(*args: Array) -> Array:
 
         ref1 = xp.where(cond, f1(*arrays), fill_value)
         ref2 = xp.where(cond, f1(*arrays), f2(*arrays))
-        # TODO remove asarrays once all backends support Array API 2024.12
-        ref3 = xp.where(cond, *asarrays(f1(*arrays), float_fill_value, xp=xp))
+        if library is Backend.ARRAY_API_STRICT:
+            # FIXME https://github.com/data-apis/array-api-strict/issues/131
+            ref3 = xp.where(cond, *asarrays(f1(*arrays), float_fill_value, xp=xp))
+        else:
+            ref3 = xp.where(cond, f1(*arrays), float_fill_value)
 
         xp_assert_close(res1, ref1, rtol=2e-16)
         xp_assert_equal(res2, ref2)