def test_discrete_legacy_state_pickling(): legacy_state = { "n": 3, } d = Discrete(1) assert "start" in d.__dict__ del d.__dict__["start"] # legacy did not include start param assert "start" not in d.__dict__ d.__setstate__(legacy_state) assert d.start == 0 assert d.n == 3
def test_space_legacy_state_pickling(): legacy_state = { "shape": ( 1, 2, 3, ), "dtype": np.int64, "np_random": np.random.default_rng(), "n": 3, } space = Discrete(1) space.__setstate__(legacy_state) assert space.shape == legacy_state["shape"] assert space._shape == legacy_state["shape"] assert space.np_random == legacy_state["np_random"] assert space._np_random == legacy_state["np_random"] assert space.n == 3 assert space.dtype == legacy_state["dtype"]