Chex 0.1.4
What's Changed
- Add an InitVar field in the dataclass tests. by @copybara-service in #161
- Download latest .pylintrc version in tests. by @copybara-service in #167
- Fix
assert_axis_dimension_comparator
usages. by @copybara-service in #168 - Update "jax.tree_util" functions by @copybara-service in #171
- Use jax.tree_util.tree_map in place of deprecated tree_multimap. by @copybara-service in #175
- Silence some pytype errors. by @copybara-service in #174
- Add chex.Dimensions utility for readable shape asserts. by @copybara-service in #169
Full Changelog: v0.1.3...v0.1.4