add any, all

Author: lucascolley

pixi.lock

@@ -10,7 +10,7 @@ name = "pint-array"
 description = "Pint quantities with array API standard arrays."
 readme = "README.md"
 dynamic = ["version"]
-requires-python = ">=3.9"
+requires-python = ">=3.10"
 
 [project.urls]
 Homepage = "https://github.com/lucascolley/pint-array"

pyproject.toml

@@ -5,8 +5,6 @@
 Pint interoperability with array API standard arrays.
 """
 
-from __future__ import annotations
-
 import importlib
 import sys
 import textwrap
@@ -328,6 +326,24 @@ def prod(x, /, *args, axis=None, **kwargs):
 
     mod.prod = prod
 
+    for func_str in (
+        "any",
+        "all",
+    ):
+
+        def func(x, /, *args, func_str=func_str, **kwargs):
+            x = asarray(x)
+            magnitude = xp.asarray(x.magnitude, copy=True)
+            if x._is_multiplicative:
+                xp_func = getattr(xp, func_str)
+                magnitude = xp_func(magnitude, *args, **kwargs)
+                return ArrayUnitQuantity(magnitude, None)
+
+            msg = "Boolean value of Quantity with offset unit is ambiguous."
+            raise ValueError(msg)
+
+        setattr(mod, func_str, func)
+
     #  "mul": product of all units in `all_args`
     # - "delta": `first_input_units`, unless non-multiplicative,
     # which uses delta version