Update pint/facets/numpy/numpy_func.py

Co-authored-by: Justus Magin <[email protected]>
hgrecco · Nov 13, 2023 · 0977c3d · 0977c3d
1 parent 8e6be43
commit 0977c3d
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/pint/facets/numpy/numpy_func.py b/pint/facets/numpy/numpy_func.py
@@ -291,10 +291,7 @@ def implementation(*args, **kwargs):
             ]
         ):
             # the sequence may contain different units, so fall back to element-wise
-            return np.array(
-                [func(args[0][i], args[1][i]) for i in range(len(args[0]))],
-                dtype=object,
-            )
+            return np.array([func(*func_args) for func_args in zip(*args)], dtype=object)
 
         first_input_units = _get_first_input_units(args, kwargs)
         if input_units == "all_consistent":