Fix for gym version 0.9.5. (#1471)

This commit is contained in:
Robert Nishihara 2018-01-25 13:58:15 -08:00 committed by Richard Liaw
parent ab5d4a6010
commit f6c835e4b8

View file

@ -78,7 +78,6 @@ class AtariRamPreprocessor(Preprocessor):
class OneHotPreprocessor(Preprocessor):
def _init(self):
assert self._obs_space.shape == ()
self.shape = (self._obs_space.n,)
def transform(self, observation):