Merge pull request #626 from ArnoStrouwen/destats

deprecate destats in extensions
SciML · Feb 17, 2024 · 3b7832e · 3b7832e
2 parents eca78b8 + 1fa8e29
commit 3b7832e
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 6 deletions.
diff --git a/ext/SciMLBaseChainRulesCoreExt.jl b/ext/SciMLBaseChainRulesCoreExt.jl
@@ -43,7 +43,7 @@ function ChainRulesCore.rrule(config::ChainRulesCore.RuleConfig{
                 typeof(VA.alg),
                 typeof(VA.interp),
                 typeof(VA.alg_choice),
-                typeof(VA.destats),
+                typeof(VA.stats),
             }(du,
                 nothing,
                 nothing,
@@ -54,7 +54,7 @@ function ChainRulesCore.rrule(config::ChainRulesCore.RuleConfig{
                 VA.interp,
                 VA.dense,
                 0,
-                VA.destats,
+                VA.stats,
                 VA.alg_choice,
                 VA.retcode)
             (NoTangent(), Δ′, NoTangent(), NoTangent())

diff --git a/ext/SciMLBaseZygoteExt.jl b/ext/SciMLBaseZygoteExt.jl
@@ -24,8 +24,8 @@ using RecursiveArrayTools
         N = length(VA.prob.p)
         Δ′ = ODESolution{T, N, typeof(du), Nothing, Nothing, typeof(VA.t),
             typeof(VA.k), typeof(dprob), typeof(VA.alg), typeof(VA.interp),
-            typeof(VA.destats), typeof(VA.alg_choice)}(du, nothing, nothing,
-            VA.t, VA.k, dprob, VA.alg, VA.interp, VA.dense, 0, VA.destats,
+            typeof(VA.stats), typeof(VA.alg_choice)}(du, nothing, nothing,
+            VA.t, VA.k, dprob, VA.alg, VA.interp, VA.dense, 0, VA.stats,
             VA.alg_choice, VA.retcode)
         (Δ′, nothing, nothing)
     end
@@ -53,8 +53,8 @@ end
         N = length(VA.prob.p)
         Δ′ = ODESolution{T, N, typeof(du), Nothing, Nothing, typeof(VA.t),
             typeof(VA.k), typeof(dprob), typeof(VA.alg), typeof(VA.interp),
-            typeof(VA.destats), typeof(VA.alg_choice)}(du, nothing, nothing,
-            VA.t, VA.k, dprob, VA.alg, VA.interp, VA.dense, 0, VA.destats,
+            typeof(VA.stats), typeof(VA.alg_choice)}(du, nothing, nothing,
+            VA.t, VA.k, dprob, VA.alg, VA.interp, VA.dense, 0, VA.stats,
             VA.alg_choice, VA.retcode)
         (Δ′, nothing, nothing)
     end