@@ -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