1
0
Fork 0
mirror of https://gitlab.rlp.net/mobitar/ReCo.jl.git synced 2024-11-08 22:21:08 +00:00

Specify trajectory types

This commit is contained in:
Mo8it 2022-01-11 18:43:43 +01:00
parent 0cf59e04a5
commit eba1e53cd2

View file

@ -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 end
function run_rl(; function run_rl(;