Add missing sympy operators for boolean logic

MilesCranmer · Sep 19, 2023 · ff2f93a · ff2f93a
1 parent 6c92e1c
commit ff2f93a
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/pysr/export_sympy.py b/pysr/export_sympy.py
@@ -47,6 +47,8 @@
     "ceil": sympy.ceiling,
     "sign": sympy.sign,
     "gamma": sympy.gamma,
+    "logical_or": lambda x, y: sympy.Piecewise((1.0, 0.0), ((x > 0) | (y > 0), True)),
+    "logical_and": lambda x, y: sympy.Piecewise((1.0, 0.0), ((x > 0) & (y > 0), True)),
 }