All notable changes to this project will be documented in this file.
The format is based on Keep a Changelog, and this project adheres to Semantic Versioning.
- Add method
PyTreeSpec.one_level
andPyTreeSpec.is_one_level
by @XuehaiPan in #179. - Add method
PyTreeSpec.transform
by @XuehaiPan in #177.
- Mark some arguments as positional-only as of Python 3.8+ by @XuehaiPan in #178.
- Drop Python 3.7 support by @XuehaiPan in #161.
0.13.1 - 2024-11-12
- Upload coverage / JUnit results / core dumps in CI workflows by @XuehaiPan in #170 and #172.
- Add more info to
tree_flatten_one_level
by @XuehaiPan in #168. - Improve typing support for generic
PyTree[T]
and registry lookup / register functions by @XuehaiPan in #160 and #166.
- Move include directory
include/{ => optree}/*.h
by @XuehaiPan in #167.
- Improve typing support for
optree.dataclasses.dataclass
andoptree.dataclasses.field
by @manulari in #165.
0.13.0 - 2024-10-03
- Add Python 3.13t support by @XuehaiPan in #137.
- Expose Python implementation for C utilities for
namedtuple
andPyStructSequence
by @XuehaiPan in #157. - Add
dataclasses
integration by @XuehaiPan in #142. - Add Python 3.13 support by @XuehaiPan in #156.
- Respect cmake variable
pybind11_DIR
by @XuehaiPan in #155. - Add tests with PyDebug enabled in CI by @XuehaiPan in #150.
- Split implementation files and add more
inline
/constexpr
/noexcept
qualifiers by @XuehaiPan in #159. - Use
cmake
'sFindPython
module by @XuehaiPan in #151.
- Fix potential segmentation fault for
structseq_fields
cache support by @XuehaiPan in #150.
0.12.1 - 2024-07-06
- Fix warning regression during import when launch with strict warning filters by @XuehaiPan in #149.
0.12.0 - 2024-07-05
- Add context manager to temporarily set the dictionary sorting mode by @XuehaiPan in #147.
- Add PyPy support by @XuehaiPan in #145.
- Add 32-bit wheels for Linux and Windows by @XuehaiPan in #141.
- Add Linux ppc64le and s390x wheels by @XuehaiPan in #138.
- Add accessor APIs
tree_flatten_with_accessor
andPyTreeSpec.accessors
by @XuehaiPan in #108. - Add submodule
optree.functools
by @XuehaiPan in #134.
- Use
stable
tag instead of 2.12.0 forpybind11
version by @XuehaiPan in #146. - Refactor the raw import statement in
setup.py
withimportlib
utilities by @XuehaiPan in #135. - Update minimal version of
typing-extensions
to 4.5.0 fortyping_extensions.deprecated
by @XuehaiPan in #134. - Update string representation for
OrderedDict
by @XuehaiPan in #133.
- Fix gc for self-referential case by implementing
tp_traverse
by @XuehaiPan in #144. - Fix potential segmentation fault for pickling support by @XuehaiPan in #143.
- Update CI runner image for Python 3.7 on macOS by @XuehaiPan in #135.
- Deprecate key path APIs by @XuehaiPan in #108.
- Deprecate
optree.Partial
and replace withoptree.functools.partial
by @XuehaiPan in #134.
0.11.0 - 2024-03-26
- Add function
is_namedtuple_instance
andis_structseq_instance
and result caches by @XuehaiPan in #121. - Add
tree_iter
function by @XuehaiPan in #130. - Add API to unregister node type in the registry by @XuehaiPan in #124.
- Add tree map functions with transposed outputs
tree_transpose_map
andtree_transpose_map_with_path
by @XuehaiPan in #127. - Add static constructors to create
PyTreeSpec
instances by @XuehaiPan in #120. - Cache intermediate
str
objects inPyObject_GetAttr
calls by @XuehaiPan in #106 and #109. - Install
clang-format
andclang-tidy
from PyPI by @XuehaiPan in #107. - Also check
_make
and_asdict
in functionis_namedtuple_class
by @XuehaiPan in #105.
- Set recursion limit to 1000 for all platforms by @XuehaiPan in #121.
- Allow types to be registered in both the global namespace and custom namespaces by @XuehaiPan in #124.
- Set
treespec_is_leaf
as strict by default by @XuehaiPan in #120. - Reorder functions for better code correspondence between C++ and Python by @XuehaiPan in #117.
- Standardize
py::handle
andpy::object
usage in function signature by @XuehaiPan in #115. - Reorder cases for
namedtuple
andPyStructSequence
types by @XuehaiPan in #111. - Use
__bases__
rather than__base__
in functionis_structseq_class
by @XuehaiPan in #104.
- Fix potential segmentation fault when modifying
treespec.entries()
by @XuehaiPan in #116.
0.10.0 - 2023-11-07
- Add
tree_ravel
function for JAX/NumPy/PyTorch array/tensor tree manipulation by @XuehaiPan in #100. - Expose node kind enum for
PyTreeSpec
by @XuehaiPan in #98. - Expose function
tree_flatten_one_level
by @XuehaiPan in #101. - Add tree broadcast functions
broadcast_common
,tree_broadcast_common
,tree_broadcast_map
, andtree_broadcast_map_with_path
by @XuehaiPan in #87. - Add function
tree_is_leaf
and addis_leaf
argument to functionall_leaves
by @XuehaiPan in #93. - Add methods
PyTreeSpec.entry
andPyTreeSpec.child
by @XuehaiPan in #88. - Add Python 3.12 support by @XuehaiPan in #90.
- Allow passing third-party dependency version from environment variable by @XuehaiPan in #80.
- Set recursion limit to 2000 for all platforms by @XuehaiPan in #97.
- Make
PyTreeSpec.is_prefix
to be consistent withPyTreeSpec.flatten_up_to
by @XuehaiPan in #94. - Decrease the
MAX_RECURSION_DEPTH
to 2000 on Windows by @XuehaiPan in #85. - Bump
abseil-cpp
version to 20230802.1 by @XuehaiPan in #80.
- Memorize ongoing
repr
/hash
calls to resolve infinite recursion under self-referential case by @XuehaiPan and @JieRen98 in #82.
- Remove dependence on
abseil-cpp
by @XuehaiPan in #85.
0.9.2 - 2023-09-18
- Bump
pybind11
version to 2.11.1 and add initial Python 3.12 support by @XuehaiPan in #78. - Bump
abseil-cpp
version to 20230802.0 by @XuehaiPan in #79.
- Fix empty paths when flatten with custom
is_leaf
function by @XuehaiPan in #76.
0.9.1 - 2023-05-23
- Use
py::type::handle_of(obj)
rather than deprecatedobj.get_type()
by @XuehaiPan in #49. - Bump
abseil-cpp
version to 20230125.3 by @XuehaiPan in #57.
- Add
@runtime_checkable
decorator forCustomTreeNode
protocol class by @XuehaiPan in #56.
0.9.0 - 2023-03-23
- Preserve dict key order in the output of
tree_unflatten
,tree_map
, andtree_map_with_path
by @XuehaiPan in #46.
- Change keyword argument
initializer
back toinitial
fortree_reduce
to align withfunctools.reduce
C implementation by @XuehaiPan in #47.
0.8.0 - 2023-03-14
- Add methods
PyTreeSpec.paths
andPyTreeSpec.entries
by @XuehaiPan in #43. - Allow tree-map with mixed inputs of ordered and unordered dictionaries by @XuehaiPan in #42.
- Add more utility functions for
namedtuple
andPyStructSequence
type by @XuehaiPan in #41. - Add methods
PyTreeSpec.is_prefix
andPyTreeSpec.is_suffix
and functiontree_broadcast_prefix
by @XuehaiPan in #40. - Add tree reduce functions
tree_sum
,tree_max
, andtree_min
by @XuehaiPan in #39. - Test dict key equality with
PyDict_Contains
($O (n)$) rather than sorting ($O (n \log n)$) by @XuehaiPan in #37. - Make error message more clear when value mismatch by @XuehaiPan in #36.
- Add
ruff
andflake8
plugins integration by @XuehaiPan in #33 and #34.
- Allow tree-map with mixed inputs of ordered and unordered dictionaries by @XuehaiPan in #42.
- Use more appropriate exception handling (e.g., change
ValueError
toTypeError
instructseq_fields
) by @XuehaiPan in #41. - Inherit
optree._C.InternalError
fromSystemError
rather thanRuntimeError
by @XuehaiPan in #41. - Change keyword argument
initial
toinitializer
fortree_reduce
to align withfunctools.reduce
by @XuehaiPan in #39.
0.7.0 - 2023-02-07
- Add
PyStructSequence
types as internal node types by @XuehaiPan in #30.
- Add
PyStructSequence
types as internal node types by @XuehaiPan in #30. - Use postponed evaluation of annotations by @XuehaiPan in #28.
0.6.0 - 2023-02-02
- Add Linux AArch64 and Windows ARM64 wheels by @XuehaiPan in #25.
- Add property
PyTreeSpec.type
and methodPyTreeSpec.is_leaf
by @XuehaiPan in #26. - Raise a warning when registering subclasses of
namedtuple
by @XuehaiPan in #24. - Add
clang-tidy
integration and update code style by @XuehaiPan in #20.
- Add
doctest
integration and fix docstring by @XuehaiPan in #23.
0.5.1 - 2023-01-21
- Add property
PyTreeSpec.num_children
by @XuehaiPan. - Update docstring and documentation by @XuehaiPan.
0.5.0 - 2022-11-30
- Add custom exceptions for internal error handling by @XuehaiPan.
- Fix
PyTreeSpec
equality test and hash by @XuehaiPan.
0.4.2 - 2022-11-27
- Better internal error handling by @XuehaiPan in #17.
- Use static raw pointers for global imports by @XuehaiPan in #16.
0.4.1 - 2022-11-25
- Fix segmentation fault error for global imports @XuehaiPan in #14.
0.4.0 - 2022-11-25
- Add namespace support for custom node type registry by @XuehaiPan in #12.
- Add tree flatten and tree map functions with extra paths by @XuehaiPan in #11.
- Add in-place version of tree-map function
tree_map_
by @XuehaiPan. - Add macOS ARM64 wheels by @XuehaiPan in #9.
- Add Python 3.11 support by @XuehaiPan.
- Use shallow clone for third-party Git repos by @XuehaiPan.
- Use cmake FetchContent rather than Git submodules by @XuehaiPan.
- Drop Python 3.6 support by @XuehaiPan.
0.3.0 - 2022-10-26
- Add Read the Docs integration by @XuehaiPan.
- Add benchmark script and results by @XuehaiPan.
- Support both "
None
is Node" and "None
is Leaf" by @XuehaiPan. - Add
OrderedDict
anddefaultdict
anddeque
as builtin support by @XuehaiPan.
- Reorganize code structure and rename
PyTreeDef
toPyTreeSpec
by @XuehaiPan.
- Fix Python 3.6 support by @XuehaiPan.
- Fix generic
NamedTuple
for Python 3.8-3.10 by @XuehaiPan. - Fix builds for Python 3.8-3.10 on Windows by @XuehaiPan.
0.2.0 - 2022-09-24
- Add
cibuildwheel
integration for building wheels by @XuehaiPan. - Add full type annotations by @XuehaiPan.
- Improve custom tree node representation by @XuehaiPan.
- Fix cross-platform compatibility by @XuehaiPan.
0.1.0 - 2022-09-16
- The first beta release of OpTree by @XuehaiPan.
- OpTree with Linux / Windows / macOS x64 support by @XuehaiPan.