fix test

JuliaReinforcementLearning · Oct 10, 2023 · 3b6ac40 · 3b6ac40
1 parent 5019b7f
commit 3b6ac40
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/ReinforcementLearningCore/src/utils/networks.jl b/src/ReinforcementLearningCore/src/utils/networks.jl
@@ -112,7 +112,7 @@ function (model::GaussianNetwork)(state, action)
     x = model.pre(state)
     μ, raw_σ = model.μ(x), model.σ(x)
     σ = clamp.(raw_σ, model.min_σ, model.max_σ)
-    logp_π = diagnormlogpdf(μ, σ, inversesquash(model.squash).(action)) .+ logpdfcorrection(z, typeof(model.squash))
+    logp_π = diagnormlogpdf(μ, σ, inversesquash(model.squash).(action)) .+ logpdfcorrection(inversesquash(model.squash).(action), typeof(model.squash))
     return logp_π
 end