[BugFix] Fix deepcopy of TensorDictParams (#580)

pytorch · Nov 27, 2023 · 1a9aca2 · 1a9aca2
1 parent c730421
commit 1a9aca2
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/tensordict/nn/params.py b/tensordict/nn/params.py
@@ -548,9 +548,12 @@ def __ne__(self, other: object) -> TensorDictBase:
         ...
 
     def __getattr__(self, item: str) -> Any:
-        try:
-            return getattr(self.__dict__["_param_td"], item)
-        except AttributeError:
+        if not item.startswith("_"):
+            try:
+                return getattr(self.__dict__["_param_td"], item)
+            except AttributeError:
+                return super().__getattr__(item)
+        else:
             return super().__getattr__(item)
 
     @_fallback