Spaces:

MilesCranmer
/

PySR

Running

MilesCranmer commited on Mar 1, 2021

Commit

afef357

•

1 Parent(s): 6decb44

Fix some jax function equivalencies

Files changed (1) hide show

pysr/export.py CHANGED Viewed

@@ -45,8 +45,8 @@ try:
         # Note: May raise error for ints and complexes
         sympy.erf: "jsp.erf",
         sympy.erfc: "jsp.erfc",
-        sympy.LessThan: "jnp.le",
-        sympy.GreaterThan: "jnp.ge",
         sympy.And: "jnp.logical_and",
         sympy.Or: "jnp.logical_or",
         sympy.Not: "jnp.logical_not",

         # Note: May raise error for ints and complexes
         sympy.erf: "jsp.erf",
         sympy.erfc: "jsp.erfc",
+        sympy.LessThan: "jnp.less",
+        sympy.GreaterThan: "jnp.greater",
         sympy.And: "jnp.logical_and",
         sympy.Or: "jnp.logical_or",
         sympy.Not: "jnp.logical_not",