MilesCranmer commited on
Commit
018c3a1
1 Parent(s): df788e5

refactor: sympy typing issue

Browse files
Files changed (1) hide show
  1. pysr/export_sympy.py +2 -2
pysr/export_sympy.py CHANGED
@@ -30,8 +30,8 @@ sympy_mappings = {
30
  "acosh": lambda x: sympy.acosh(x),
31
  "acosh_abs": lambda x: sympy.acosh(abs(x) + 1),
32
  "asinh": sympy.asinh,
33
- "atanh": lambda x: sympy.atanh(sympy.Mod(x + 1, 2) - 1),
34
- "atanh_clip": lambda x: sympy.atanh(sympy.Mod(x + 1, 2) - 1),
35
  "abs": abs,
36
  "mod": sympy.Mod,
37
  "erf": sympy.erf,
 
30
  "acosh": lambda x: sympy.acosh(x),
31
  "acosh_abs": lambda x: sympy.acosh(abs(x) + 1),
32
  "asinh": sympy.asinh,
33
+ "atanh": lambda x: sympy.atanh(sympy.Mod(x + 1, 2) - sympy.S(1)),
34
+ "atanh_clip": lambda x: sympy.atanh(sympy.Mod(x + 1, 2) - sympy.S(1)),
35
  "abs": abs,
36
  "mod": sympy.Mod,
37
  "erf": sympy.erf,