Fix the idxs error

nathanaelbosch · Jun 9, 2024 · b6913f0 · b6913f0
1 parent b66ad29
commit b6913f0
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/src/solution.jl b/src/solution.jl
@@ -288,6 +288,7 @@ function (interp::ODEFilterPosterior)(
     p,
     continuity,
 ) where {deriv}
+    d = interp.cache.d
     q = interp.cache.q
     dv = deriv.parameters[1]
     proj = if deriv == Val{0}
@@ -301,8 +302,11 @@ function (interp::ODEFilterPosterior)(
         t, interp.ts, interp.x_filt, interp.x_smooth, interp.diffusions, interp.cache;
         smoothed=interp.smooth)
     u = proj * x
-    @assert length(u) == length(idxs)
-    return Gaussian(u.μ, u.Σ)
+    P = zeros(Bool, length(idxs), d)
+    for (i, idx) in enumerate(idxs)
+        P[i, idx] = 1
+    end
+    return P * u
 end
 function (interp::ODEFilterPosterior)(
     t::AbstractVector{<:Real},