4da0e542d54c7977aae5bda5242cab15ab3f344b,rllib/tests/test_supported_spaces.py,,,#,183

Before Change

        ACTION_SPACES_TO_TEST = {
            "discrete": Discrete(5),
        }
        OBSERVATION_SPACES_TO_TEST = {
            "vector": Box(0.0, 1.0, (5, ), dtype=np.float32),
            "atari": Box(0.0, 1.0, (210, 160, 3), dtype=np.float32),
        }

    sys.exit(pytest.main(["-v", __file__]))

After Change

    // One can specify the specific TestCase class to run.
    // None for all unittest.TestCase classes in this file.
    class_ = sys.argv[1] if len(sys.argv) > 0 else None
    sys.exit(pytest.main(
        ["-v", __file__ + ("" if class_ is None else "::" + class_)]))

Instances