Chex 0.0.8
Changes:
- Add support for static_broadcasted_argnums to fake_pmap;
- Allows sets of alternatives and ellipsis in assert_shape;
- Format @variant test names to use only underscores and lowercase letters;
- Fix incorrect type annotation in asserts.py;
- Fix dataclass (un-)flatten functions;
- Add more tests for dataclasses;
- Raise ValueError when no variants are selected;
- Exclude chex' internal frames from AssertionError tracebacks;
- Add '[Chex] ' prefix to AssertionError messages;
- Include path to leaves that failed the equality check in
assert_tree_all_close; - Clean up asserts.py;
- Asserts which only make sense on >1 tree now demand this (can result in breakages in the existing code).