Fix pre-commit

Farama-Foundation · Jan 18, 2024 · b3ab650 · b3ab650
1 parent ef2cf68
commit b3ab650
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/test/test_vector/test_render.py b/test/test_vector/test_render.py
@@ -1,7 +1,8 @@
 from pettingzoo.butterfly import pistonball_v6
-import supersuit as ss
 from stable_baselines3.common.vec_env import VecVideoRecorder
 
+import supersuit as ss
+
 
 def schedule(episode_idx):
     print(episode_idx)
@@ -13,7 +14,7 @@ def make_sb3_record_env():
     print(env.render_mode)
     env = ss.pettingzoo_env_to_vec_env_v1(env)
     envs = ss.concat_vec_envs_v1(env, 1, num_cpus=0, base_class="stable_baselines3")
-    envs = VecVideoRecorder(envs, f"/tmp", schedule)
+    envs = VecVideoRecorder(envs, "/tmp", schedule)
     return envs