From bb9a52c15eb758a5a3d6db360ea3f46825cfca18 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Cl=C3=A9ment=20Dumas?= Date: Wed, 17 Jan 2024 14:41:51 +0100 Subject: [PATCH] reformat --- test/test_vector/test_render.py | 9 +++------ 1 file changed, 3 insertions(+), 6 deletions(-) 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)