fix: use of numpy in jax expressions

MilesCranmer · Jun 22, 2024 · 723dbce · 723dbce
1 parent 148b3a7
commit 723dbce
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/pysr/export_jax.py b/pysr/export_jax.py
@@ -1,3 +1,4 @@
+import numpy as np  # noqa: F401
 import sympy
 
 # Special since need to reduce arguments.