perf: Handle unique roots only

AMICI-dev · Dec 12, 2023 · fd227ec · fd227ec
1 parent e0fbeec
commit fd227ec
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/python/sdist/amici/de_export.py b/python/sdist/amici/de_export.py
@@ -2754,7 +2754,7 @@ def _process_heavisides(
         heavisides = []
         # run through the expression tree and get the roots
         tmp_roots_old = self._collect_heaviside_roots(dt_expanded.args)
-        for tmp_old in tmp_roots_old:
+        for tmp_old in set(tmp_roots_old):
             # we want unique identifiers for the roots
             tmp_new = self._get_unique_root(tmp_old, roots)
             heavisides.append((sp.Heaviside(tmp_old), tmp_new))