amend

pytorch · Jul 2, 2024 · 5a854e9 · 5a854e9
1 parent ff0101d
commit 5a854e9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/test/test_env.py b/test/test_env.py
@@ -3225,7 +3225,7 @@ def test_partial_rest(self, batched):
             env = SerialEnv(2, [env0, env1])
         s = env.reset()
         i = 0
-        for i in range(10):
+        for i in range(10):  # noqa: B007
             s, s_ = env.step_and_maybe_reset(
                 s.set("action", torch.ones(2, 1, dtype=torch.int))
             )