Translate log2 and log10 to their direct sympy equivalents instead of…

… log(x)/log(base)
MilesCranmer · Sep 4, 2024 · dee1356 · dee1356
1 parent 339cc0a
commit dee1356
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pysr/export_sympy.py b/pysr/export_sympy.py
@@ -39,8 +39,8 @@
     "erf": sympy.erf,
     "erfc": sympy.erfc,
     "log": lambda x: sympy.log(x),
-    "log10": lambda x: sympy.log(x, 10),
-    "log2": lambda x: sympy.log(x, 2),
+    "log10": lambda x: sympy.codegen.cfunctions.log10(x),
+    "log2": lambda x: sympy.codegen.cfunctions.log2(x),
     "log1p": lambda x: sympy.log(x + 1),
     "log_abs": lambda x: sympy.log(abs(x)),
     "log10_abs": lambda x: sympy.log(abs(x), 10),