[BugFix] NonTensor should not convert anything to numpy

ghstack-source-id: 6006ff9b7edde96f785a599e30945c1c2d53fa97 Pull Request resolved: #2771
pytorch · Feb 7, 2025 · 20b7d3d · 20b7d3d
1 parent de2a96e
commit 20b7d3d
Showing 1 changed file with 12 additions and 0 deletions.
diff --git a/torchrl/data/tensor_specs.py b/torchrl/data/tensor_specs.py
@@ -2619,6 +2619,18 @@ def unbind(self, dim: int = 0):
             for i in range(self.shape[dim])
         )
 
+    def to_numpy(
+        self, val: torch.Tensor | TensorDictBase, safe: bool = None
+    ) -> np.ndarray | dict:
+        return val
+
+    def encode(
+        self,
+        val: np.ndarray | torch.Tensor | TensorDictBase,
+        *,
+        ignore_device: bool = False,
+    ) -> torch.Tensor | TensorDictBase:
+        return val
 
 class _UnboundedMeta(abc.ABCMeta):
     def __call__(cls, *args, **kwargs):