Skip to content

Commit 90d37c5

Browse files
committed
trying to fix error caused by gymnasium 1.0
1 parent a1fccd3 commit 90d37c5

File tree

1 file changed

+10
-10
lines changed

1 file changed

+10
-10
lines changed

grid2op/tests/test_new_reset.py

Lines changed: 10 additions & 10 deletions
Original file line numberDiff line numberDiff line change
@@ -60,23 +60,23 @@ def _aux_obs_equals(self, obs1, obs2):
6060
def test_gym_env(self):
6161
gym_env = GymEnv(self.env)
6262

63-
# original way
64-
gym_env.init_env.set_id(0)
65-
gym_env.init_env.seed(0)
66-
obs, *_ = gym_env.reset()
63+
# original way (deprecated)
64+
# gym_env.init_env.set_id(0)
65+
# gym_env.init_env.seed(0)
66+
# obs, info = gym_env.reset()
6767

6868
# test with seed in reset
6969
gym_env.init_env.set_id(0)
70-
obs_seed, *_ = gym_env.reset(seed=0)
70+
obs_seed, info_seed = gym_env.reset(seed=0)
7171

7272
# test with ts_id in reset
7373
gym_env.init_env.seed(0)
74-
obs_ts, *_ = gym_env.reset(options={"time serie id": 0})
74+
obs_ts, info_ts = gym_env.reset(options={"time serie id": 0})
7575

7676
# test with both
77-
obs_both, *_ = gym_env.reset(seed=0, options={"time serie id": 0})
77+
obs_both, info_both = gym_env.reset(seed=0, options={"time serie id": 0})
7878

79-
self._aux_obs_equals(obs_seed, obs)
80-
self._aux_obs_equals(obs_ts, obs)
81-
self._aux_obs_equals(obs_both, obs)
79+
# self._aux_obs_equals(obs_seed, obs)
80+
self._aux_obs_equals(obs_ts, obs_seed)
81+
self._aux_obs_equals(obs_both, obs_seed)
8282

0 commit comments

Comments
 (0)