gryf/coach
1
0
Fork 0
mirror of https://github.com/gryf/coach.git synced 2025-12-17 19:20:19 +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/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):