diff --git a/test/test_vector/test_render.py b/test/test_vector/test_render.py index 6a3a842..b6898c0 100644 --- a/test/test_vector/test_render.py +++ b/test/test_vector/test_render.py @@ -8,7 +8,7 @@ def schedule(episode_idx): return episode_idx <= 1 -def make_record_env(): +def make_sb3_record_env(): env = pistonball_v6.parallel_env(render_mode="rgb_array") print(env.render_mode) env = ss.pettingzoo_env_to_vec_env_v1(env) @@ -17,17 +17,14 @@ def make_record_env(): return envs -def record_video_test(): - envs = make_record_env() +def test_record_video_sb3(): + envs = make_sb3_record_env() envs.reset() for _ in range(100): envs.step([envs.action_space.sample() for _ in range(envs.num_envs)]) envs.close() -record_video_test() - - # def make_env(): # env = pistonball_v6.parallel_env(render_mode="rgb_array") # env = ss.pettingzoo_env_to_vec_env_v1(env)