Highlights
TensorDict 0.14.1 is a focused patch release with correctness fixes for module metadata, lazy and non-tensor indexing, and frozen tensorclass deserialization. It contains no breaking changes, new features, or deprecations.
Bug Fixes
- Added informative representations for
TensorDictModuleBasesubclasses, including their child modules and input and output keys (#1767). - Preserved lazy structure when batched indexing selects ragged entries from a
LazyStackedTensorDict(#1772). - Kept
TensorDictModuleoutput-key metadata synchronized whenout_keysis reassigned (#1773). - Restored pickling, unpickling, and shallow copying for frozen tensorclasses (#1775).
- Made advanced indexing of
NonTensorStackreturn independent entries, preventing indexed results from mutating their source or one another (#1778).
Installation
pip install tensordict==0.14.1Contributors
Thanks to @priba, @gtnv, @simonschlaepfer, and @vmoens for their contributions to this release.
Full Changelog: v0.14.0...v0.14.1