1
0
mirror of https://github.com/gryf/coach.git synced 2025-12-18 19:50:17 +01:00

fix for measurements in the initial state (fix for DFP)

This commit is contained in:
Gal Leibovich
2018-05-29 16:47:38 +03:00
parent 7725dabc86
commit 2807c29f27

View File

@@ -371,7 +371,7 @@ class Agent(object):
'observation': observation 'observation': observation
} }
if self.tp.agent.use_measurements: 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'] self.curr_state['measurements'] = self.env.state['measurements']
else: else:
self.curr_state['measurements'] = np.zeros(0) self.curr_state['measurements'] = np.zeros(0)