From 86362683b1f24b3b2aa04fb8d1d98048eb7c32ee Mon Sep 17 00:00:00 2001 From: Zach Dwiel Date: Tue, 20 Feb 2018 11:39:20 -0500 Subject: [PATCH] comment --- agents/agent.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/agents/agent.py b/agents/agent.py index 66085aa..b88c662 100644 --- a/agents/agent.py +++ b/agents/agent.py @@ -378,9 +378,10 @@ class Agent(object): state['observation'] = stack_observation([], state['observation'], self.tp.env.observation_stack_size) self.curr_state = state - # TODO: this should be handled in the environment if self.tp.agent.use_measurements: + # TODO: this should be handled in the environment self.curr_state['measurements'] = self.env.measurements + if self.tp.agent.use_accumulated_reward_as_measurement: self.curr_state['measurements'] = np.append(self.curr_state['measurements'], 0)