Add SHAIL

This commit is contained in:
ebuehrle
2022-02-17 23:49:06 +01:00
parent 9de6bfe9a3
commit 1624e1a349
11 changed files with 79 additions and 98 deletions

View File

@@ -9,6 +9,10 @@ class TransformObservation(gym.wrappers.TransformObservation):
def __getattr__(self, name):
return getattr(self.env, name)
class OptionsTimeLimit(gym.wrappers.TimeLimit):
def __getattr__(self, name):
return getattr(self.env, name)
class CollisionPenaltyWrapper(Wrapper):
def __init__(self, env, collision_distance, collision_penalty, *args, **kwargs):