Fix tdir handling in FakeIntegrator

SciML · Mar 12, 2024 · 3471ad7 · 3471ad7
1 parent 9dba71c
commit 3471ad7
Showing 1 changed file with 5 additions and 0 deletions.
diff --git a/src/callback_tracking.jl b/src/callback_tracking.jl
@@ -168,6 +168,11 @@ struct FakeIntegrator{uType, P, tType, tprevType}
     tprev::tprevType
 end
 
+function Base.getproperty(fi::FakeIntegrator, s::Symbol)
+    s === :tdir && return sign(fi.t - fi.tprev)
+    return getfield(fi, s)
+end
+
 struct CallbackSensitivityFunction{fType, Alg <: AbstractOverloadingSensitivityAlgorithm,
     C <: AdjointDiffCache, pType} <: SensitivityFunction
     f::fType