[Versioning] v0.6.2 fix

pytorch · Nov 14, 2024 · cdd5cb3 · cdd5cb3
1 parent 0b3c778
commit cdd5cb3
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tensordict/base.py b/tensordict/base.py
@@ -10607,7 +10607,7 @@ def to(tensor):
             else:
                 apply_kwargs["device"] = device if device is not None else self.device
                 apply_kwargs["batch_size"] = batch_size
-                apply_kwargs["out"] = self if inplace else None
+                apply_kwargs["out"] = None
                 apply_kwargs["checked"] = True
                 if non_blocking_pin:
 

diff --git a/tensordict/utils.py b/tensordict/utils.py
@@ -2779,7 +2779,7 @@ def _rebuild_njt_from_njt(x, values, offsets, lengths):
         tgt = mb_unwrap_functional_tensor(new_thing)
         src = mb_unwrap_functional_tensor(ragged_source)
         tgt.nested_int_memo = src.nested_int_memo
-    else:
+    elif new_thing is not None:
         _tensor_symint_registry[new_thing] = _tensor_symint_registry[ragged_source]
 
     return NestedTensor(