MilesCranmer commited on
Commit
723dbce
·
unverified ·
1 Parent(s): 148b3a7

fix: use of numpy in jax expressions

Browse files
Files changed (1) hide show
  1. pysr/export_jax.py +1 -0
pysr/export_jax.py CHANGED
@@ -1,3 +1,4 @@
 
1
  import sympy
2
 
3
  # Special since need to reduce arguments.
 
1
+ import numpy as np # noqa: F401
2
  import sympy
3
 
4
  # Special since need to reduce arguments.