MilesCranmer commited on
Commit
5dbc3b7
1 Parent(s): 09bfff6

Add missing `greater` operator in sympy mapping

Browse files
Files changed (1) hide show
  1. pysr/export_sympy.py +1 -0
pysr/export_sympy.py CHANGED
@@ -50,6 +50,7 @@ sympy_mappings = {
50
  "round": lambda x: sympy.ceiling(x - 0.5),
51
  "max": lambda x, y: sympy.Piecewise((y, x < y), (x, True)),
52
  "min": lambda x, y: sympy.Piecewise((x, x < y), (y, True)),
 
53
  "cond": lambda x, y: sympy.Piecewise((y, x > 0), (0.0, True)),
54
  "logical_or": lambda x, y: sympy.Piecewise((1.0, (x > 0) | (y > 0)), (0.0, True)),
55
  "logical_and": lambda x, y: sympy.Piecewise((1.0, (x > 0) & (y > 0)), (0.0, True)),
 
50
  "round": lambda x: sympy.ceiling(x - 0.5),
51
  "max": lambda x, y: sympy.Piecewise((y, x < y), (x, True)),
52
  "min": lambda x, y: sympy.Piecewise((x, x < y), (y, True)),
53
+ "greater": lambda x, y: sympy.Piecewise((1.0, x > y), (0.0, True)),
54
  "cond": lambda x, y: sympy.Piecewise((y, x > 0), (0.0, True)),
55
  "logical_or": lambda x, y: sympy.Piecewise((1.0, (x > 0) | (y > 0)), (0.0, True)),
56
  "logical_and": lambda x, y: sympy.Piecewise((1.0, (x > 0) & (y > 0)), (0.0, True)),