make integral functions callable

SciML · Sep 21, 2023 · 740576a · 740576a
1 parent c0f4062
commit 740576a
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/scimlfunctions.jl b/src/scimlfunctions.jl
@@ -2397,6 +2397,8 @@ TruncatedStacktraces.@truncate_stacktrace BatchIntegralFunction 1 2
 (f::ODEFunction)(args...) = f.f(args...)
 (f::NonlinearFunction)(args...) = f.f(args...)
 (f::IntervalNonlinearFunction)(args...) = f.f(args...)
+(f::IntegralFunction)(args...) = f.f(args...)
+(f::BatchIntegralFunction)(args...) = f.f(args...)
 
 function (f::DynamicalODEFunction)(u, p, t)
     ArrayPartition(f.f1(u.x[1], u.x[2], p, t), f.f2(u.x[1], u.x[2], p, t))