From 52560a2aaef08c85f5e2156f23f6abc5c03d1cc1 Mon Sep 17 00:00:00 2001 From: Zach Dwiel Date: Thu, 4 Oct 2018 17:01:21 -0400 Subject: [PATCH] introduce property GraphManager.current_step_counter --- rl_coach/graph_managers/graph_manager.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/rl_coach/graph_managers/graph_manager.py b/rl_coach/graph_managers/graph_manager.py index 6325d28..d942ede 100644 --- a/rl_coach/graph_managers/graph_manager.py +++ b/rl_coach/graph_managers/graph_manager.py @@ -309,7 +309,7 @@ class GraphManager(object): self.reset_internal_state(force_environment_reset=True) # act for at least steps, though don't interrupt an episode - count_end = self.total_steps_counters[self.phase][EnvironmentSteps] + steps.num_steps + count_end = self.total_steps_counters[self.phase][steps.__class__] + steps.num_steps while self.total_steps_counters[self.phase][steps.__class__] < count_end: self.act(EnvironmentEpisodes(1))