From 2807c29f27a0547bbc1d7d380478846d47245273 Mon Sep 17 00:00:00 2001 From: Gal Leibovich Date: Tue, 29 May 2018 16:47:38 +0300 Subject: [PATCH] fix for measurements in the initial state (fix for DFP) --- agents/agent.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/agents/agent.py b/agents/agent.py index 010d5d8..888f1b2 100644 --- a/agents/agent.py +++ b/agents/agent.py @@ -371,7 +371,7 @@ class Agent(object): 'observation': observation } if self.tp.agent.use_measurements: - if 'measurements' in self.curr_state.keys(): + if 'measurements' in self.env.state.keys(): self.curr_state['measurements'] = self.env.state['measurements'] else: self.curr_state['measurements'] = np.zeros(0)