add module to rgf call

SciML · Sep 14, 2023 · fb66ff2 · fb66ff2
1 parent 5fff919
commit fb66ff2
Showing 1 changed file with 10 additions and 10 deletions.
diff --git a/src/systems/pde/pdesystem.jl b/src/systems/pde/pdesystem.jl
@@ -108,16 +108,16 @@ struct PDESystem <: ModelingToolkit.AbstractMultivariateSystem
                 throw(ArgumentError("The number of analytic solutions must match the number of dependent variables"))
             end
 
-            if isnothing(analytic_func)
-                analytic_func = map(analytic) do eq
-                    args = arguments(eq.lhs)
-                    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(ex))
-                end
-            end
-        end
+			if isnothing(analytic_func)
+				analytic_func = map(analytic) do eq
+					args = arguments(eq.lhs)
+					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__, @__MODULE__, ex))
+				end
+			end
+		end
 
         if !isnothing(analytic_func)
             analytic_func = analytic_func isa Dict ? analytic_func : analytic_func |> Dict