gryf/coach
1
0
Fork 0
mirror of https://github.com/gryf/coach.git synced 2025-12-18 19:50:17 +01:00
Code Issues Projects Releases Wiki Activity

integration test changes to reach the train part (#254)

Browse Source 
* integration test changes to override heatup to 1000 steps +  run each preset for 30 sec (to make sure we reach the train part)

* fixes to failing presets uncovered with this change + changes in the golden testing to properly test BatchRL

* fix for rainbow dqn

* fix to gym_environment (due to a change in Gym 0.12.1) + fix for rainbow DQN + some bug-fix in utils.squeeze_list

* fix for NEC agent
This commit is contained in:
Gal Leibovich
2019-03-27 21:14:19 +02:00
committed by GitHub
parent 6e08c55ad5
commit 310d31c227
8 changed files with 28 additions and 17 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
rl_coach/tests/presets/test_presets.py
View File

@@ -57,14 +57,16 @@ def test_preset_runs(preset):
experiment_name = ".test-" + preset
params = ["python3", "rl_coach/coach.py", "-p", preset, "-ns", "-e", experiment_name]
# overriding heatup steps to some small number of steps (1000), so to finish the heatup stage, and get to train
params = ["python3", "rl_coach/coach.py", "-p", preset, "-ns", "-e", experiment_name, '-cp',
'heatup_steps=EnvironmentSteps(1000)']
if level != "":
params += ["-lvl", level]
p = Popen(params)
# wait 10 seconds overhead of initialization etc.
time.sleep(10)
# wait 30 seconds overhead of initialization, and finishing heatup.
time.sleep(30)
return_value = p.poll()
if return_value is None: