Guard scalar arguments with API_VERSION>=2024.12

Author: ev-br

array_api_strict/_helpers.py

@@ -2,11 +2,20 @@
 """
 import numpy as np
 
+from ._flags import get_array_api_strict_flags
+
+
 _py_scalars = (bool, int, float, complex)
 
+
 def _maybe_normalize_py_scalars(x1, x2):
     from ._array_object import Array
 
+    flags = get_array_api_strict_flags()
+    if flags["api_version"] < "2024.12": # XXX: string comparison for versions
+        # scalars will fail at the call site
+        return x1, x2
+
     if isinstance(x1, _py_scalars):
         if isinstance(x2, _py_scalars):
             raise TypeError(f"Two scalars not allowed, {type(x1) = } and {type(x2) =}")

array_api_strict/tests/test_elementwise_functions.py

@@ -218,6 +218,10 @@ def _sample_scalar(category):
         else:
             raise ValueError(f'Unknown {category = }')
 
+    # Use the latest version of the standard so that scalars are actually allowed
+    with pytest.warns(UserWarning):
+        set_array_api_strict_flags(api_version="2024.12")
+
     for func_name, types in elementwise_function_input_types.items():
         dtypes = _dtype_categories[types]
         func = getattr(_elementwise_functions, func_name)