Skip to content
This repository was archived by the owner on May 6, 2021. It is now read-only.

Commit 43a010d

Browse files
authored
in _run, add hook for PRE_EXPERIMENT_STAGE & Post... (#219)
1 parent cfbd624 commit 43a010d

File tree

1 file changed

+2
-0
lines changed

1 file changed

+2
-0
lines changed

src/core/run.jl

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,7 @@ function check(policy, env) end
1515

1616
function _run(policy::AbstractPolicy, env::AbstractEnv, stop_condition, hook::AbstractHook)
1717

18+
hook(PRE_EXPERIMENT_STAGE, policy, env)
1819
is_stop = false
1920
while !is_stop
2021
reset!(env)
@@ -43,5 +44,6 @@ function _run(policy::AbstractPolicy, env::AbstractEnv, stop_condition, hook::Ab
4344
hook(POST_EPISODE_STAGE, policy, env)
4445
end
4546
end
47+
hook(POST_EXPERIMENT_STAGE, policy, env)
4648
hook
4749
end

0 commit comments

Comments
 (0)