From f6c835e4b8e9ddff3aaedfced40f4d5232a3f0dc Mon Sep 17 00:00:00 2001 From: Robert Nishihara Date: Thu, 25 Jan 2018 13:58:15 -0800 Subject: [PATCH] Fix for gym version 0.9.5. (#1471) --- python/ray/rllib/models/preprocessors.py | 1 - 1 file changed, 1 deletion(-) diff --git a/python/ray/rllib/models/preprocessors.py b/python/ray/rllib/models/preprocessors.py index 01e9db16f..f6ec1fa5e 100644 --- a/python/ray/rllib/models/preprocessors.py +++ b/python/ray/rllib/models/preprocessors.py @@ -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):