dodox/fancy_gym
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fix: Minor bugs in time aware obs wrapper

Browse Source
This commit is contained in:
Dominik Moritz Roth 2023-06-11 13:47:38 +02:00
parent e44b0ed9ed
commit 2ad42f4132
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
fancy_gym/utils/time_aware_observation.py
View File

@ -28,10 +28,10 @@ class TimeAwareObservation(gym.ObservationWrapper, gym.utils.RecordConstructorAr
assert env.observation_space.__class__ in allowed_classes, str(env.observation_space)+' is not supported. Only Box or Dict'
low = np.append(env.observation_space.low, 0.0)
high = np.append(env.observation_space.high, 1.0)
if env.observation_space.__class__ in [Box, OldBox]:
low = np.append(env.observation_space.low, 0.0)
high = np.append(env.observation_space.high, 1.0)
self.observation_space = Box(low, high, dtype=dtype)
else:
import pdb
@ -49,7 +49,7 @@ class TimeAwareObservation(gym.ObservationWrapper, gym.utils.RecordConstructorAr
Returns:
The observation with the time step appended to (relative to total number of steps)
"""
return np.append(observation, self.t / getattr(self.env, '_max_episode_steps'))
return np.append(observation, self.t / self.env.spec.max_episode_steps)
def step(self, action):
"""Steps through the environment, incrementing the time step.