Merge pull request #590 from MilesCranmer/missing-sympy

Add missing `greater` operator in sympy mapping
MilesCranmer · Mar 31, 2024 · 46227e8 · 46227e8
2 parents 09bfff6 + 5dbc3b7
commit 46227e8
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/pysr/export_sympy.py b/pysr/export_sympy.py
@@ -50,6 +50,7 @@
     "round": lambda x: sympy.ceiling(x - 0.5),
     "max": lambda x, y: sympy.Piecewise((y, x < y), (x, True)),
     "min": lambda x, y: sympy.Piecewise((x, x < y), (y, True)),
+    "greater": lambda x, y: sympy.Piecewise((1.0, x > y), (0.0, True)),
     "cond": lambda x, y: sympy.Piecewise((y, x > 0), (0.0, True)),
     "logical_or": lambda x, y: sympy.Piecewise((1.0, (x > 0) | (y > 0)), (0.0, True)),
     "logical_and": lambda x, y: sympy.Piecewise((1.0, (x > 0) & (y > 0)), (0.0, True)),