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

reenable redis; better error message

Browse Source
This commit is contained in:
Zach Dwiel
2018-09-17 19:50:03 +00:00
committed by zach dwiel
parent 009cf670f3
commit 3328b25549
3 changed files with 11 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files

11
rl_coach/rollout_worker.py
View File

@@ -40,7 +40,11 @@ def wait_for_checkpoint(checkpoint_dir, timeout=10):
if has_checkpoint(checkpoint_dir):
return
raise ValueError('checkpoint never found in {checkpoint_dir}'.format(
raise ValueError((
'Waited {timeout} seconds, but checkpoint never found in'
' {checkpoint_dir}'
).format(
timeout=timeout,
checkpoint_dir=checkpoint_dir,
))
@@ -82,9 +86,8 @@ def main():
graph_manager = short_dynamic_import(expand_preset(args.preset), ignore_module_case=True)
# TODO: get this working, this expects that memory already has a redis ip and port
# graph_manager.agent_params.memory.redis_ip = args.redis_ip
# graph_manager.agent_params.memory.redis_port = args.redis_port
graph_manager.agent_params.memory.redis_ip = args.redis_ip
graph_manager.agent_params.memory.redis_port = args.redis_port
rollout_worker(
graph_manager=graph_manager,