amend

pytorch · Jan 27, 2024 · aebb0fa · aebb0fa
1 parent e193148
commit aebb0fa
Showing 1 changed file with 1 addition and 3 deletions.
diff --git a/test/test_tensordict.py b/test/test_tensordict.py
@@ -3026,9 +3026,7 @@ def test_named_apply_complete(self, td_name, device):
         def count(name, value, keys):
             keys.add(name)
 
-        td.named_apply(
-            functools.partial(count, keys=keys_complete), nested_keys=True
-        )
+        td.named_apply(functools.partial(count, keys=keys_complete), nested_keys=True)
         td.named_apply(
             functools.partial(count, keys=keys_not_complete), nested_keys=False
         )