hotfix new extensions

SciML · Nov 3, 2023 · 8b5e58d · 8b5e58d
1 parent db00ab8
commit 8b5e58d
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/Project.toml b/Project.toml
@@ -1,7 +1,7 @@
 name = "SciMLBase"
 uuid = "0bca4576-84f4-4d90-8ffe-ffa030f20462"
 authors = ["Chris Rackauckas <[email protected]> and contributors"]
-version = "2.7.0"
+version = "2.7.1"
 
 [deps]
 ADTypes = "47edcb42-4c32-4615-8424-f2b9edc5f35b"

diff --git a/ext/SciMLBaseChainRulesCoreExt.jl b/ext/SciMLBaseChainRulesCoreExt.jl
@@ -1,5 +1,6 @@
 module SciMLBaseChainRulesCoreExt
 
+using SciMLBase
 import ChainRulesCore
 import ChainRulesCore: NoTangent, @non_differentiable
 
@@ -118,7 +119,7 @@ function ChainRulesCore.rrule(::Type{
     SDESolution{uType, tType, isinplace, P, NP, F, G, K, ND}(u, args...), SDESolutionAdjoint
 end
 
-function ChainRulesCore.rrule(::DiffEqBase.EnsembleSolution, sim, time, converged)
+function ChainRulesCore.rrule(::SciMLBase.EnsembleSolution, sim, time, converged)
     out = EnsembleSolution(sim, time, converged)
     function EnsembleSolution_adjoint(p̄::AbstractArray{T, N}) where {T, N}
         arrarr = [[p̄[ntuple(x -> Colon(), Val(N - 2))..., j, i]