From eba1e53cd29ff4aba7fc026d99bb086b212e267a Mon Sep 17 00:00:00 2001 From: Mo8it Date: Tue, 11 Jan 2022 18:43:43 +0100 Subject: [PATCH] Specify trajectory types --- src/RL/RL.jl | 6 +++++- 1 file changed, 5 insertions(+), 1 deletion(-) diff --git a/src/RL/RL.jl b/src/RL/RL.jl index bf7250a..2ad332a 100644 --- a/src/RL/RL.jl +++ b/src/RL/RL.jl @@ -298,7 +298,11 @@ function gen_agent(n_states::Int64, n_actions::Int64, ϵ_stable::Float64) ), ) - return Agent(; policy=policy, trajectory=VectorSARTTrajectory()) + trajectory = VectorSARTTrajectory(; + state=Int64, action=Int64, reward=Float64, terminal=Bool + ) + + return Agent(; policy=policy, trajectory=trajectory) end function run_rl(;