Possible speedup for unit_math.sum function.

f6d90066 · Jason Swails · 9c38d980 · f6d90066
Commit f6d90066 authored Oct 06, 2014 by Jason Swails
Show whitespace changes
Inline Side-by-side

Showing with 4 additions and 0 deletions

wrappers/python/simtk/unit/unit_math.py wrappers/python/simtk/unit/unit_math.py +4 -0

No files found.
--- a/wrappers/python/simtk/unit/unit_math.py
+++ b/wrappers/python/simtk/unit/unit_math.py
@@ -156,6 +156,10 @@ def sum(val):
    >>> sum((2.0*meter, 30.0*centimeter))
    Quantity(value=2.3, unit=meter)
    """
+    try:
+        return val.sum()
+    except AttributeError:
+        pass
    if len(val) == 0:
        return 0
    result = val[0]