ps1 -> ps2.

JuliaGNI · May 15, 2024 · d6b98b7 · d6b98b7
1 parent b381814
commit d6b98b7
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/linear_symplectic_attention.jl b/test/linear_symplectic_attention.jl
@@ -18,7 +18,7 @@ function test_application_of_lsa(n::Integer=4, seq_length::Integer=5, T=Float64)
     # test for Array as input
     arr = rand(T, 2 * n, seq_length)
     @test l₁(arr, ps₁) ≈ vcat(arr[1:n, :] + arr[(n + 1):(2 * n), :] * ps₁.A, arr[(n + 1):(2 * n), :])
-    @test l₂(arr, ps₂) ≈ vcat(arr[1:n, :], arr[(n + 1):(2 * n), :] + arr[1:n, :] * ps₁.A)
+    @test l₂(arr, ps₂) ≈ vcat(arr[1:n, :], arr[(n + 1):(2 * n), :] + arr[1:n, :] * ps₂.A)
 end
 
 test_application_of_lsa()