amend

pytorch · Oct 9, 2023 · 39aabdb · 39aabdb
1 parent bc070b1
commit 39aabdb
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/tensordict/nn/params.py b/tensordict/nn/params.py
@@ -424,7 +424,7 @@ def apply(
         ...
 
     @_unlock_and_set(inplace=True)
-    def _apply(*args, **kwargs):
+    def _apply_nest(*args, **kwargs):
         ...
 
     @_get_post_hook

diff --git a/tensordict/tensordict.py b/tensordict/tensordict.py
@@ -5231,7 +5231,9 @@ def _empty_like(td: T, *args, **kwargs) -> T:
             "cloned, preventing empty_like to be called. "
             "Consider calling tensordict.to_tensordict() first."
         ) from err
-    return tdclone._fast_apply(lambda x: torch.empty_like(x, *args, **kwargs), inplace=True)
+    return tdclone._fast_apply(
+        lambda x: torch.empty_like(x, *args, **kwargs), inplace=True
+    )
 
 
 @implements_for_td(torch.clone)