Chex 0.1.82
What's Changed
Chex has moved from external dm-tree
to the JAX-native jax.tree_util
library for operating with PyTree
s.
This means that None
s will not be treated as separate leaves by chex tree assertions any longer.
Note that you can still check for None
s using chex.assert_tree_no_nones()
.
- Remove usage of dm-tree in asserts. by @copybara-service in #288
- Add support for
kw_only
for chex.dataclass. by @jblespiau in #290 - Deprecate
ignore_nones
. by @copybara-service in #292
Full Changelog: v0.1.81...v0.1.82