def test_gym_preprocessors(self): p1 = ModelCatalog.get_preprocessor(gym.make("CartPole-v0")) self.assertEqual(type(p1), NoPreprocessor) p2 = ModelCatalog.get_preprocessor(gym.make("FrozenLake-v0")) self.assertEqual(type(p2), OneHotPreprocessor) p3 = ModelCatalog.get_preprocessor(gym.make("MsPacman-ram-v0")) self.assertEqual(type(p3), AtariRamPreprocessor) p4 = ModelCatalog.get_preprocessor(gym.make("MsPacmanNoFrameskip-v4")) self.assertEqual(type(p4), GenericPixelPreprocessor)
def test_tuple_preprocessor(self): class TupleEnv: def __init__(self): self.observation_space = Tuple( [Discrete(5), Box(0, 5, shape=(3, ), dtype=np.float32)]) pp = ModelCatalog.get_preprocessor(TupleEnv()) self.assertTrue(isinstance(pp, TupleFlatteningPreprocessor)) self.assertEqual(pp.shape, (8, )) self.assertEqual(list(pp.transform((0, np.array([1, 2, 3])))), [float(x) for x in [1, 0, 0, 0, 0, 1, 2, 3]])