Skip to content

Chex 0.0.8

Choose a tag to compare

@hbq1 hbq1 released this 02 Jul 18:08

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).