From 5dbc3b7b39d39a097cd94140793247acd8cb257f Mon Sep 17 00:00:00 2001 From: Miles Cranmer Date: Sun, 31 Mar 2024 15:24:19 +0100 Subject: [PATCH] Add missing `greater` operator in sympy mapping --- pysr/export_sympy.py | 1 + 1 file changed, 1 insertion(+) diff --git a/pysr/export_sympy.py b/pysr/export_sympy.py index f09ee2a70..3bc49e5ad 100644 --- 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)),