1
0
mirror of https://github.com/gryf/coach.git synced 2025-12-17 19:20:19 +01:00

Ignoring redis sub if testing

This commit is contained in:
Ajay Deshpande
2018-10-08 12:01:12 -07:00
committed by zach dwiel
parent 7f00235ed5
commit 0e121c5762
3 changed files with 6 additions and 4 deletions

View File

@@ -376,7 +376,7 @@ class GraphManager(object):
if self.agent_params.memory.memory_backend_params.run_type == "worker":
data_store = get_data_store(self.data_store_params)
data_store.load_from_store()
# perform several steps of playing
result = None
@@ -435,7 +435,7 @@ class GraphManager(object):
if steps.num_steps > 0:
self.phase = RunPhase.TRAIN
self.reset_internal_state(force_environment_reset=True)
#TODO - the below while loop should end with full episodes, so to avoid situations where we have partial
# TODO - the below while loop should end with full episodes, so to avoid situations where we have partial
# episodes in memory
count_end = self.total_steps_counters[self.phase][steps.__class__] + steps.num_steps
while self.total_steps_counters[self.phase][steps.__class__] < count_end: