gryf/coach
1
0
Fork 0
mirror of https://github.com/gryf/coach.git synced 2025-12-18 11:40:18 +01:00
Code Issues Projects Releases Wiki Activity

multiple bug fixes in dealing with measurements + CartPole_DFP preset (#92)

Browse Source
This commit is contained in:
Itai Caspi
2018-04-23 10:44:46 +03:00
committed by GitHub
parent 5d5562bf62
commit 52eb159f69
5 changed files with 31 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
environments/doom_environment_wrapper.py
View File

@@ -135,7 +135,7 @@ class DoomEnvironmentWrapper(EnvironmentWrapper):
# extract all data from the current state
state = self.game.get_state()
if state is not None and state.screen_buffer is not None:
self.observation = {
self.state = {
'observation': state.screen_buffer,
'measurements': state.game_variables,
}

2
environments/gym_environment_wrapper.py
View File

@@ -113,7 +113,7 @@ class GymEnvironmentWrapper(EnvironmentWrapper):
self.timestep_limit = self.env.spec.timestep_limit
else:
self.timestep_limit = None
self.measurements_size = len(self.step(0)['info'].keys())
self.measurements_size = (len(self.step(0)['info'].keys()),)
self.random_initialization_steps = self.tp.env.random_initialization_steps
def _wrap_state(self, state):