From 7382a142bb1ea1d86efa70de750bdb2cb09e7265 Mon Sep 17 00:00:00 2001 From: Zach Dwiel Date: Tue, 2 Oct 2018 15:29:57 -0400 Subject: [PATCH] remove unused steps parameter from GraphManager.train --- rl_coach/graph_managers/graph_manager.py | 4 ++-- rl_coach/training_worker.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/rl_coach/graph_managers/graph_manager.py b/rl_coach/graph_managers/graph_manager.py index 48081cb..65645ff 100644 --- a/rl_coach/graph_managers/graph_manager.py +++ b/rl_coach/graph_managers/graph_manager.py @@ -325,7 +325,7 @@ class GraphManager(object): [environment.handle_episode_ended() for environment in self.environments] - def train(self, steps: TrainingSteps) -> None: + def train(self) -> None: """ Perform several training iterations for all the levels in the hierarchy :param steps: number of training iterations to perform @@ -426,7 +426,7 @@ class GraphManager(object): # The actual steps being done on the environment are decided by the agents themselves. # This is just an high-level controller. self.act(EnvironmentSteps(1)) - self.train(TrainingSteps(1)) + self.train() self.occasionally_save_checkpoint() self.phase = RunPhase.UNDEFINED diff --git a/rl_coach/training_worker.py b/rl_coach/training_worker.py index 922ae72..4031cd9 100644 --- a/rl_coach/training_worker.py +++ b/rl_coach/training_worker.py @@ -42,7 +42,7 @@ def training_worker(graph_manager, checkpoint_dir, policy_type): steps += 1 graph_manager.phase = core_types.RunPhase.TRAIN - graph_manager.train(core_types.TrainingSteps(1)) + graph_manager.train() graph_manager.phase = core_types.RunPhase.UNDEFINED if steps * graph_manager.agent_params.algorithm.num_consecutive_playing_steps.num_steps > graph_manager.steps_between_evaluation_periods.num_steps * eval_offset: