[BugFix] densify non tensor stack is a no-op

ghstack-source-id: edbc22ce562cd918ce5dd5c0441e47cdadf7d88a Pull Request resolved: #1194
pytorch · Jan 30, 2025 · f3794ba · f3794ba
1 parent 9b66338
commit f3794ba
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/tensordict/tensorclass.py b/tensordict/tensorclass.py
@@ -3562,6 +3562,10 @@ def _from_list(cls, datalist: List, device: torch.device, ndim: int | None = Non
             stack_dim=0,
         )
 
+    def densify(self, layout: torch.layout = torch.strided):
+        # No need to do anything with a non tensor stack
+        return self
+
     def update(
         self,
         input_dict_or_td: dict[str, CompatibleType] | T,