fix observed for DEintegrator

src/integrator_interface.jl

@@ -426,12 +426,6 @@ end
 
 # SymbolicIndexingInterface
 SymbolicIndexingInterface.symbolic_container(A::DEIntegrator) = A.f
-function SymbolicIndexingInterface.is_observed(A::DEIntegrator, sym)
-    return !is_variable(A, sym) && !is_parameter(A, sym) && !is_independent_variable(A, sym) && symbolic_type(sym) == ScalarSymbolic()
-end
-function SymbolicIndexingInterface.observed(A::DEIntegrator, sym)
-    return getobserved(A)(sym)
-end
 SymbolicIndexingInterface.parameter_values(A::DEIntegrator) = A.p
 SymbolicIndexingInterface.state_values(A::DEIntegrator) = A.u
 SymbolicIndexingInterface.current_time(A::DEIntegrator) = A.t

src/problems/problem_interface.jl

@@ -6,12 +6,7 @@
 end
 
 SymbolicIndexingInterface.symbolic_container(prob::AbstractSciMLProblem) = prob.f
-function SymbolicIndexingInterface.is_observed(A::AbstractSciMLProblem, sym)
-    return !is_variable(A, sym) && !is_parameter(A, sym) && !is_independent_variable(A, sym) && symbolic_type(sym) == ScalarSymbolic()
-end
-function SymbolicIndexingInterface.observed(A::AbstractSciMLProblem, sym)
-    return getobserved(A)(sym)
-end
+
 SymbolicIndexingInterface.parameter_values(prob::AbstractSciMLProblem) = prob.p
 SymbolicIndexingInterface.state_values(prob::AbstractSciMLProblem) = prob.u0
 SymbolicIndexingInterface.current_time(prob::AbstractSciMLProblem) = prob.tspan[1]
@@ -33,7 +28,7 @@
         elseif is_independent_variable(prob.f, sym)
             return getindepsym(prob)
         elseif is_observed(prob.f, sym)
-            obs = SymbolicIndexingInterface.observed(prob.f, sym)
+            obs = SymbolicIndexingInterface.observed(prob, sym)
             if is_time_dependent(prob.f)
                 return obs(prob.u0, prob.p, 0.0)
             else

src/scimlfunctions.jl

@@ -2324,7 +2324,7 @@
             typeof(_colorvec),
             typeof(sys)}(_f, mass_matrix, analytic, tgrad, jac,
             jvp, vjp, jac_prototype, sparsity, Wfact,
-            Wfact_t, W_prototype, paramjac, 
+            Wfact_t, W_prototype, paramjac,
             observed, _colorvec, sys)
     else
         ODEFunction{iip, specialize,
@@ -2336,7 +2336,7 @@
             typeof(_colorvec),
             typeof(sys)}(_f, mass_matrix, analytic, tgrad, jac,
             jvp, vjp, jac_prototype, sparsity, Wfact,
-            Wfact_t, W_prototype, paramjac, 
+            Wfact_t, W_prototype, paramjac,
             observed, _colorvec, sys)
     end
 end
@@ -2848,7 +2848,7 @@
             typeof(f.mass_matrix), typeof(f.analytic), typeof(f.tgrad),
             typeof(f.jac), typeof(f.jvp), typeof(f.vjp), typeof(f.jac_prototype),
             typeof(f.sparsity), typeof(f.Wfact), typeof(f.Wfact_t),
-            typeof(f.paramjac), typeof(f.ggprime), 
+            typeof(f.paramjac), typeof(f.ggprime),
             typeof(f.observed), typeof(f.colorvec), typeof(f.sys)}(newf, newg,
             f.mass_matrix,
             f.analytic,
@@ -2943,7 +2943,7 @@
             typeof(colorvec),
             typeof(sys)}(f1, f2, g, mass_matrix, _func_cache, analytic,
             tgrad, jac, jvp, vjp, jac_prototype, sparsity,
-            Wfact, Wfact_t, paramjac, 
+            Wfact, Wfact_t, paramjac,
             observed, colorvec, sys)
     end
 end
@@ -3120,7 +3120,7 @@
             typeof(analytic), typeof(tgrad),
             typeof(jac), typeof(jvp), typeof(vjp), typeof(jac_prototype),
             typeof(sparsity), typeof(Wfact), typeof(Wfact_t),
-            typeof(paramjac), 
+            typeof(paramjac),
             typeof(observed), typeof(_colorvec),
             typeof(sys)}(_f, mass_matrix, analytic, tgrad,
             jac, jvp, vjp, jac_prototype, sparsity,
@@ -3473,7 +3473,7 @@
             jvp, vjp, jac_prototype,
             sparsity, Wfact,
             Wfact_t,
-            paramjac, ggprime, 
+            paramjac, ggprime,
             observed, _colorvec, sys)
     end
 end
@@ -3566,7 +3566,7 @@
             typeof(_f), typeof(mass_matrix), typeof(analytic), typeof(tgrad),
             typeof(jac), typeof(jvp), typeof(vjp), typeof(jac_prototype),
             typeof(sparsity), typeof(Wfact),
-            typeof(Wfact_t), typeof(paramjac), 
+            typeof(Wfact_t), typeof(paramjac),
             typeof(observed),
             typeof(_colorvec), typeof(sys), typeof(resid_prototype)}(_f, mass_matrix,
             analytic, tgrad, jac,
@@ -3935,7 +3935,7 @@
 has_Wfact_t(f::AbstractSciMLFunction) = __has_Wfact_t(f) && f.Wfact_t !== nothing
 has_paramjac(f::AbstractSciMLFunction) = __has_paramjac(f) && f.paramjac !== nothing
 has_sys(f::AbstractSciMLFunction) = __has_sys(f) && f.sys !== nothing
-function has_syms(f::AbstractSciMLFunction) 
+function has_syms(f::AbstractSciMLFunction)
   if __has_syms(f)
     f.syms !== nothing
   else
@@ -4051,9 +4051,21 @@
 function SymbolicIndexingInterface.observed(fn::AbstractSciMLFunction, sym)
   if has_observed(fn)
     if is_time_dependent(fn)
-      return (u, p, t) -> fn.observed(sym, u, p, t)
+      return if hasmethod(fn.observed, Tuple{typeof(sym)})
+        fn.observed(sym)
+      else
+        let obs = fn.observed, sym = sym
+          (u, p, t) -> obs(sym, u, p, t)
+        end
+      end
     else
-      return (u, p) -> fn.observed(sym, u, p)
+      return if hasmethod(fn.observed, Tuple{typeof(sym)})
+        fn.observed(sym)
+      else
+        let obs = fn.observed, sym = sym
+          (u, p) -> obs(sym, u, p)
+        end
+      end
     end
   end
   error("SciMLFunction does not have observed")

src/solutions/solution_interface.jl

@@ -47,13 +47,6 @@ SymbolicIndexingInterface.is_independent_variable(::AbstractNoTimeSolution, sym)
 
 SymbolicIndexingInterface.independent_variable_symbols(::AbstractNoTimeSolution) = []
 
-for soltype in [AbstractTimeseriesSolution, AbstractNoTimeSolution]
-    @eval function SymbolicIndexingInterface.observed(A::$(soltype), sym::Symbol)
-        has_sys(A.prob.f) || error("Cannot use observed without system")
-        return SymbolicIndexingInterface.observed(A, getproperty(A.prob.f.sys, sym))
-    end
-end
-
 SymbolicIndexingInterface.is_time_dependent(::AbstractTimeseriesSolution) = true
 
 SymbolicIndexingInterface.is_time_dependent(::AbstractNoTimeSolution) = false

test/downstream/Project.toml

@@ -6,9 +6,11 @@ NonlinearSolve = "8913a72c-1f9b-4ce2-8d82-65094dcecaec"
 Optimization = "7f7a1694-90dd-40f0-9382-eb1efda571ba"
 OptimizationOptimJL = "36348300-93cb-4f02-beb5-3c3902f8871e"
 OrdinaryDiffEq = "1dea7af3-3e70-54e6-95c3-0bf5283fa5ed"
+RecursiveArrayTools = "731186ca-8d62-57ce-b412-fbd966d074cd"
 SciMLBase = "0bca4576-84f4-4d90-8ffe-ffa030f20462"
 SciMLSensitivity = "1ed8b502-d754-442c-8d5d-10ac956f44a1"
 Sundials = "c3572dad-4567-51f8-b174-8c6c989267f4"
+SymbolicIndexingInterface = "2efcf032-c050-4f8e-a9bb-153293bab1f5"
 Unitful = "1986cc42-f94f-5a68-af5c-568840ba703d"
 Zygote = "e88e6eb3-aa80-5325-afca-941959d7151f"
 

test/downstream/symbol_indexing.jl

@@ -39,21 +39,32 @@ p = [lorenz1.σ => 10.0,
 
 tspan = (0.0, 100.0)
 prob = ODEProblem(sys_simplified, u0, tspan, p)
+integ = init(prob, Rodas4())
 sol = solve(prob, Rodas4())
 
-@test_throws Any sol[b]
-@test_throws Any sol[b, 1]
-@test_throws Any sol[b, 1:5]
-@test_throws Any sol[b, [1, 2, 3]]
-@test_throws Any sol['a']
-@test_throws Any sol['a', 1]
-@test_throws Any sol['a', 1:5]
-@test_throws Any sol['a', [1, 2, 3]]
+@testset "indexing should error" begin
+    for obj in [prob, integ, sol]
+        for sym in ['a', :b]
+            @test_throws Any obj[sym]
+            @test_throws Any obj[sym, 1]
+            @test_throws Any obj[sym, 1:5]
+            @test_throws Any obj[sym, [1, 2, 3]]
+        end
+    end
+end
+
+@testset "observed shouldn't error" begin
+    for obj in [prob, integ, sol]
+        obj[:a]
+        SymbolicIndexingInterface.observed(obj, :a)
+    end
+end
+
 
 @test sol[a] isa AbstractVector
 @test sol[:a] == sol[a]
 @test sol[a, 1] isa Real
-@test sol[:a, 1] == sol[a, 1]
+@test sol[:a, 1] == sol[a, 1] == prob[a] == prob[:a] == integ[a] == integ[:a] == -1.0
 @test sol[a, 1:5] isa AbstractVector
 @test sol[:a, 1:5] == sol[a, 1:5]
 @test sol[a, [1, 2, 3]] isa AbstractVector