mirror of
https://github.com/gryf/coach.git
synced 2025-12-17 11:10:20 +01:00
integration test changes to reach the train part (#254)
* 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:
@@ -220,7 +220,7 @@ def force_list(var):
|
||||
|
||||
|
||||
def squeeze_list(var):
|
||||
if len(var) == 1:
|
||||
if type(var) == list and len(var) == 1:
|
||||
return var[0]
|
||||
else:
|
||||
return var
|
||||
|
||||
Reference in New Issue
Block a user