Chex 0.1.86
What's Changed
- Avoid use of deprecated
device_buffer
attriutes of jax.Array by @copybara-service in #325 - Update test.sh by @copybara-service in #331
- Update
chex.assert_type
to check concrete types instead of just asserting that the type is a floating/integer sub-type. by @copybara-service in #333 - Adjust test tolerances for test_assert_trees_all_close by @copybara-service in #334
- Register dataclass at definition so that unpickling the dataclass object, type, and treedef works in most cases (with one exception). by @copybara-service in #336
- Add warnings in chex. by @copybara-service in #337
- Add Dimensions.size to get flat size from named shape. by @copybara-service in #339
- Increase version number to make a new release. by @copybara-service in #342
Full Changelog: v0.1.85...v0.1.86