updated for gym 0.25.1

This commit is contained in:
Fabian
2022-08-18 09:04:38 +02:00
parent a5cac2dd0d
commit bdea3badff
10 changed files with 93 additions and 110 deletions
+2 -2
View File
@@ -86,8 +86,8 @@ class BlackBoxWrapper(gym.ObservationWrapper):
velocity = get_numpy(self.traj_gen.get_traj_vel())
# Remove first element of trajectory as this is the current position and velocity
trajectory = trajectory[1:]
velocity = velocity[1:]
# trajectory = trajectory[1:]
# velocity = velocity[1:]
return trajectory, velocity
+2 -5
View File
@@ -1,4 +1,3 @@
from abc import abstractmethod
from typing import Union, Tuple
import gym
@@ -23,7 +22,6 @@ class RawInterfaceWrapper(gym.Wrapper):
return np.ones(self.env.observation_space.shape[0], dtype=bool)
@property
@abstractmethod
def current_pos(self) -> Union[float, int, np.ndarray, Tuple]:
"""
Returns the current position of the action/control dimension.
@@ -32,10 +30,9 @@ class RawInterfaceWrapper(gym.Wrapper):
it should, however, be implemented regardless.
E.g. The joint positions that are directly or indirectly controlled by the action.
"""
raise NotImplementedError()
raise NotImplementedError
@property
@abstractmethod
def current_vel(self) -> Union[float, int, np.ndarray, Tuple]:
"""
Returns the current velocity of the action/control dimension.
@@ -44,7 +41,7 @@ class RawInterfaceWrapper(gym.Wrapper):
it should, however, be implemented regardless.
E.g. The joint velocities that are directly or indirectly controlled by the action.
"""
raise NotImplementedError()
raise NotImplementedError
@property
def dt(self) -> float: