lint

pytorch · Feb 15, 2024 · 70c1523 · 70c1523
1 parent 2d01867
commit 70c1523
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/test/test_env.py b/test/test_env.py
@@ -2539,7 +2539,9 @@ def make_env(seed, device=device):
         return env
 
     serial_env = SerialEnv(
-        2, [functools.partial(make_env, seed=0), functools.partial(make_env, seed=seed)], device=device
+        2,
+        [functools.partial(make_env, seed=0), functools.partial(make_env, seed=seed)],
+        device=device,
     )
     r_serial = serial_env.rollout(10, policy)
 
@@ -2549,7 +2551,9 @@ def make_env(seed, device=device):
     torch.testing.assert_close(g, g_serial)
 
     p_env = ParallelEnv(
-        2, [functools.partial(make_env, seed=0), functools.partial(make_env, seed=seed)], device=device,
+        2,
+        [functools.partial(make_env, seed=0), functools.partial(make_env, seed=seed)],
+        device=device,
     )
     try:
         r_parallel = p_env.rollout(10, policy)