From 8be980912c9d5f07dc204a84f1eb853cae10a79f Mon Sep 17 00:00:00 2001 From: Zach Dwiel Date: Wed, 3 Oct 2018 16:53:52 -0400 Subject: [PATCH] fixed typo from earlier commit --- rl_coach/graph_managers/graph_manager.py | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/rl_coach/graph_managers/graph_manager.py b/rl_coach/graph_managers/graph_manager.py index 0622bc4..a8a1f5b 100644 --- a/rl_coach/graph_managers/graph_manager.py +++ b/rl_coach/graph_managers/graph_manager.py @@ -375,8 +375,7 @@ class GraphManager(object): # perform several steps of playing result = None - initial_count = self.total_steps_counters[self.phase][EnvironmentSteps] - count_end = initial_count + steps.num_steps + count_end = self.total_steps_counters[self.phase][steps.__class__] + steps.num_steps # The assumption here is that the total_steps_counters are each updated when an event # takes place (i.e. an episode ends)