eval_module

SciML · Sep 18, 2023 · 5cddadd · 5cddadd
1 parent 732c3ee
commit 5cddadd
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/systems/pde/pdesystem.jl b/src/systems/pde/pdesystem.jl
@@ -94,6 +94,7 @@ struct PDESystem <: ModelingToolkit.AbstractMultivariateSystem
         analytic = nothing,
         analytic_func = nothing,
         gui_metadata = nothing,
+		eval_module = @__MODULE__,
         checks::Union{Bool, Int} = true,
         name)
         if checks == true || (checks & CheckUnits) > 0
@@ -114,7 +115,7 @@ struct PDESystem <: ModelingToolkit.AbstractMultivariateSystem
 					p = ps isa SciMLBase.NullParameters ? [] : map(a -> a.first, ps)
 					args = vcat(DestructuredArgs(p), args)
 					ex = Func(args, [], eq.rhs) |> toexpr
-					eq.lhs => drop_expr(@RuntimeGeneratedFunction(@__MODULE__, ex))
+					eq.lhs => drop_expr(@RuntimeGeneratedFunction(eval_module, ex))
 				end
 			end
 		end