Update

[ghstack-poisoned]
pytorch · Dec 3, 2024 · 31eb86f · 31eb86f
1 parent 694f641
commit 31eb86f
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/tensordict/tensorclass.py b/tensordict/tensorclass.py
@@ -123,6 +123,8 @@ def __subclasscheck__(self, subclass):
 ]
 # Methods to be executed from tensordict, any ref to self means 'self._tensordict', no wrap of result
 _FALLBACK_METHOD_FROM_TD_NOWRAP = [
+    "_check_batch_size",
+    "_check_device",
     "_check_dim_name",
     "_check_unlock",
     "_default_get",
@@ -181,6 +183,7 @@ def __subclasscheck__(self, subclass):
     "_select",  # TODO: must be specialized
     "_set_at_tuple",
     "_set_tuple",
+    "_to_module",
     "abs",
     "abs_",
     "acos",
@@ -327,6 +330,7 @@ def __subclasscheck__(self, subclass):
     "tanh",
     "tanh_",
     "to",
+    "to_module",
     "transpose",
     "trunc",
     "trunc_",