Add mappings from sympy exact One/Half

pysr/export_jax.py

@@ -48,6 +48,9 @@ _jnp_func_lookup = {
     sympy.Max: "jnp.max",
     sympy.Min: "jnp.min",
     sympy.Mod: "jnp.mod",
+    sympy.Heaviside: "jnp.heaviside",
+    sympy.core.numbers.Half: "(lambda: 0.5)",
+    sympy.core.numbers.One: "(lambda: 1.0)",
 }
 
 

pysr/export_torch.py

@@ -77,6 +77,9 @@ def _initialize_torch():
             sympy.Max: torch.max,
             sympy.Min: torch.min,
             sympy.Mod: torch.remainder,
+            sympy.Heaviside: torch.heaviside,
+            sympy.core.numbers.Half: (lambda: 0.5),
+            sympy.core.numbers.One: (lambda: 1.0),
         }
 
         class _Node(torch.nn.Module):