mirror of
https://github.com/wassname/ray.git
synced 2026-09-09 11:32:43 +08:00
Fix for gym version 0.9.5. (#1471)
This commit is contained in:
committed by
Richard Liaw
parent
ab5d4a6010
commit
f6c835e4b8
@@ -78,7 +78,6 @@ class AtariRamPreprocessor(Preprocessor):
|
|||||||
|
|
||||||
class OneHotPreprocessor(Preprocessor):
|
class OneHotPreprocessor(Preprocessor):
|
||||||
def _init(self):
|
def _init(self):
|
||||||
assert self._obs_space.shape == ()
|
|
||||||
self.shape = (self._obs_space.n,)
|
self.shape = (self._obs_space.n,)
|
||||||
|
|
||||||
def transform(self, observation):
|
def transform(self, observation):
|
||||||
|
|||||||
Reference in New Issue
Block a user