update heaviside argument

bjodah · Sep 15, 2024 · f668160 · f668160
1 parent ebbcb8e
commit f668160
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/chempy/kinetics/rates.py b/chempy/kinetics/rates.py
@@ -198,8 +198,9 @@ def heaviside(x, z):
         elif hasattr(backend, 'clip'):  # e.g. numpy
             def heaviside(x, z):
                 assert z == 0
-                lx = backend.log(backend.maximum(x, tiny))
-                xclp = be.clip(lx, lo, hi)
+                lx = backend.log(backend.maximum(backend.array(x), tiny))
+                xclp = backend.clip(lx, lo, hi)
+                x = (xclp - lo)/(hi - lo)
                 y = (3 - 2*x)*x*x
                 return y
         else: