From 1fa8e29e04c715f267d728619e3bd89c4dcac9d5 Mon Sep 17 00:00:00 2001 From: Arno Strouwen Date: Fri, 16 Feb 2024 23:31:38 +0100 Subject: [PATCH] deprecate destats in extensions --- ext/SciMLBaseChainRulesCoreExt.jl | 4 ++-- ext/SciMLBaseZygoteExt.jl | 8 ++++---- 2 files changed, 6 insertions(+), 6 deletions(-) diff --git a/ext/SciMLBaseChainRulesCoreExt.jl b/ext/SciMLBaseChainRulesCoreExt.jl index 5cfd0fbee..b0e517502 100644 --- 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 index 1f09d55c3..77e3ea492 100644 --- 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