MilesCranmer commited on
Commit
1b10c77
1 Parent(s): 89cf586

Add missing sympy mapping for sqrtm

Browse files
Files changed (1) hide show
  1. pysr/sr.py +1 -0
pysr/sr.py CHANGED
@@ -16,6 +16,7 @@ global_extra_sympy_mappings = {}
16
  sympy_mappings = {
17
  'div': lambda x, y : x/y,
18
  'mult': lambda x, y : x*y,
 
19
  'square':lambda x : x**2,
20
  'cube': lambda x : x**3,
21
  'plus': lambda x, y : x + y,
 
16
  sympy_mappings = {
17
  'div': lambda x, y : x/y,
18
  'mult': lambda x, y : x*y,
19
+ 'sqrtm':lambda x : sympy.sqrt(abs(x)),
20
  'square':lambda x : x**2,
21
  'cube': lambda x : x**3,
22
  'plus': lambda x, y : x + y,