Skip to content

Commit 1c253aa

Browse files
committed
fixed linear Gaussian uinit test
1 parent 9cc94eb commit 1c253aa

File tree

1 file changed

+9
-9
lines changed

1 file changed

+9
-9
lines changed

test/linear-gaussian.jl

+9-9
Original file line numberDiff line numberDiff line change
@@ -29,18 +29,18 @@ end
2929
N_SAMPLES = 200
3030

3131
# Model dynamics
32-
a = 0.5
33-
b = 0.2
34-
q = 0.1
35-
E = LinearEvolution(a, Gaussian(b, q))
32+
A = 0.5
33+
B = 0.2
34+
Q = 0.1
35+
E = LinearEvolution(A, Gaussian(B, Q))
3636

3737
H = 1.0
3838
R = 0.1
3939
Obs = LinearObservationModel(H, R)
4040

41-
x0 = 0.0
41+
X0 = 0.0
4242
P0 = 1.0
43-
G0 = Gaussian(x0, P0)
43+
G0 = Gaussian(X0, P0)
4444

4545
M = LinearStateSpaceModel(E, Obs)
4646
O = LinearObservation(E, H, R)
@@ -62,8 +62,8 @@ end
6262
q::T
6363
end
6464

65-
function SSMProblems.distribution(proc::LinearGaussianDynamics; kwargs...)
66-
return Normal(convert(T, x0), convert(T, p0))
65+
function SSMProblems.distribution(proc::LinearGaussianDynamics{T}; kwargs...) where {T}
66+
return Normal(convert(T, X0), convert(T, P0))
6767
end
6868

6969
function SSMProblems.distribution(
@@ -89,7 +89,7 @@ end
8989
return StateSpaceModel(dyn, obs)
9090
end
9191

92-
lgssm = LinearGaussianStateSpaceModel(a, b, q, H, R)
92+
lgssm = LinearGaussianStateSpaceModel(A, B, Q, H, R)
9393
model = lgssm(ys)
9494

9595
@testset "PGAS" begin

0 commit comments

Comments
 (0)