Versioning

Author: Vincent Moens

.github/scripts/version_script.bat

@@ -1,5 +1,5 @@
 @echo off
-set TENSORDICT_BUILD_VERSION=0.7.1
+set TENSORDICT_BUILD_VERSION=0.7.2
 echo TENSORDICT_BUILD_VERSION is set to %TENSORDICT_BUILD_VERSION%
 
 @echo on

.github/scripts/version_script.sh

@@ -1,3 +1,3 @@
 #!/bin/bash
 
-export TENSORDICT_BUILD_VERSION=0.7.1
+export TENSORDICT_BUILD_VERSION=0.7.2

tensordict/base.py

@@ -96,7 +96,6 @@
     lazy_legacy,
     lock_blocked,
     prod,
-    set_capture_non_tensor_stack,
     set_lazy_legacy,
     strtobool,
     TensorDictFuture,
@@ -9103,8 +9102,7 @@ def newfn(item_and_out):
                     from tensordict._lazy import LazyStackedTensorDict
 
                     # We want to be able to return whichever data structure
-                    with set_capture_non_tensor_stack(False):
-                        out = LazyStackedTensorDict.maybe_dense_stack(imaplist, dim)
+                    out = LazyStackedTensorDict.maybe_dense_stack(imaplist, dim)
                 else:
                     out = torch.cat(imaplist, dim)
             return out

tensordict/tensorclass.py

@@ -3219,9 +3219,7 @@ def _stack_non_tensor(cls, list_of_non_tensor, dim=0, raise_if_non_unique=False)
 
         ids = set()
         firstdata = NO_DEFAULT
-        return_stack = not capture_non_tensor_stack()
-        if return_stack:
-            return NonTensorStack(*list_of_non_tensor, stack_dim=dim)
+        return_stack = False
         for data in list_of_non_tensor:
             if not isinstance(data, NonTensorData):
                 if raise_if_non_unique:

test/test_tensorclass.py

@@ -1258,18 +1258,16 @@ def test_pickle(self):
 
     @pytest.mark.parametrize("consolidate", [False, True])
     def test_pickle_consolidate(self, consolidate):
-        with set_capture_non_tensor_stack(False):
-
-            tc = TCStrings(a="a", b="b")
-
-            tcstack = TensorDict(tc=torch.stack([tc, tc.clone()]))
-            if consolidate:
-                tcstack = tcstack.consolidate()
-            assert isinstance(tcstack["tc"], TCStrings)
-            loaded = pickle.loads(pickle.dumps(tcstack))
-            assert isinstance(loaded["tc"], TCStrings)
-            assert loaded["tc"].a == tcstack["tc"].a
-            assert loaded["tc"].b == tcstack["tc"].b
+        tc = TCStrings(a="a", b="b")
+
+        tcstack = TensorDict(tc=torch.stack([tc, tc.clone()]))
+        if consolidate:
+            tcstack = tcstack.consolidate()
+        assert isinstance(tcstack["tc"], TCStrings)
+        loaded = pickle.loads(pickle.dumps(tcstack))
+        assert isinstance(loaded["tc"], TCStrings)
+        assert loaded["tc"].a == tcstack["tc"].a
+        assert loaded["tc"].b == tcstack["tc"].b
 
     def test_post_init(self):
         @tensorclass

test/test_tensordict.py

@@ -11210,7 +11210,6 @@ def non_tensor_data(self):
             batch_size=[],
         )
 
-    @set_capture_non_tensor_stack(False)
     def test_consolidate_nested(self):
         import pickle
 

version.txt

@@ -1 +1 @@
-0.7.1
+0.7.2