add spec property to wrapper to be forwarded to wrapped env. Fixes 1554 (#1574)

This commit is contained in:
pzhokhov
2019-06-28 15:27:43 -07:00
committed by GitHub
parent 0fb09bd9a8
commit cd7f96e4db

View File

@@ -209,13 +209,16 @@ class Wrapper(Env):
self.observation_space = self.env.observation_space self.observation_space = self.env.observation_space
self.reward_range = self.env.reward_range self.reward_range = self.env.reward_range
self.metadata = self.env.metadata self.metadata = self.env.metadata
self.spec = getattr(self.env, 'spec', None)
def __getattr__(self, name): def __getattr__(self, name):
if name.startswith('_'): if name.startswith('_'):
raise AttributeError("attempted to get missing private attribute '{}'".format(name)) raise AttributeError("attempted to get missing private attribute '{}'".format(name))
return getattr(self.env, name) return getattr(self.env, name)
@property
def spec(self):
return self.env.spec
@classmethod @classmethod
def class_name(cls): def class_name(cls):
return cls.__name__ return cls.__name__