Update src/expectation.jl

Co-authored-by: Lorenzo Van Munoz <[email protected]>
ArnoStrouwen · Feb 18, 2024 · 3bc3d4d · 3bc3d4d
1 parent c7d8d10
commit 3bc3d4d
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/src/expectation.jl b/src/expectation.jl
@@ -226,8 +226,7 @@ function integrate(quadalg, adalg::AbstractExpectationADAlgorithm, f, lb::TB, ub
                    nout = nothing, batch = nothing,
                    kwargs...) where {TB}
     #TODO check batch iip type stability w/ IntegralProblem{XXXX}
-    batch = batch==0 ? nothing : batch
-    prob = IntegralProblem(f, lb, ub, p; nout = nout, batch = batch)
+    prob = IntegralProblem(f, (lb, ub), p; nout = nout, batch = batch)
     solve(prob, quadalg; kwargs...)
 end