Skip to content

Commit 0b415fa

Browse files
Update de.py
1 parent f544004 commit 0b415fa

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

diffeqpy/de.py

+2-2
Original file line numberDiff line numberDiff line change
@@ -2,9 +2,9 @@
22
from . import load_julia_packages
33
de, _, _ = load_julia_packages("DifferentialEquations", "ModelingToolkit", "PythonCall")
44
from juliacall import Main
5-
de.jit = Main.seval("jit(x) = typeof(x).name.wrapper(ModelingToolkit.complete(ModelingToolkit.modelingtoolkitize(x); split = false), [], x.tspan)") # kinda hackey
5+
de.jit = Main.seval("global jit(x) = typeof(x).name.wrapper(ModelingToolkit.complete(ModelingToolkit.modelingtoolkitize(x); split = false), [], x.tspan)") # kinda hackey
66
de.jit32 = Main.seval("""
7-
function jit(x)
7+
global function jit(x)
88
prob = typeof(x).name.wrapper(ModelingToolkit.complete(ModelingToolkit.modelingtoolkitize(x); split = false), [], Float32.(x.tspan))
99
ModelingToolkit.remake(prob; u0 = Float32.(prob.u0), p = Float32.(prob.p))
1010
end

0 commit comments

Comments
 (0)