Chex 0.1.6

What's Changed

• Replace jnp.DeviceArray with the new public type jax.Array. by @copybara-service in #187
• Unpack assertions' kwargs at a higher level for improved readability. by @copybara-service in #209
• Allow calling @chexify-ed functions from non-main threads. by @copybara-service in #214
• Add assert_trees_all_equal_shapes_and_dtypes by @copybara-service in #219
• Implement a new unflatten for chex.dataclass that avoids init while keeping the (un)flattened order unchanged. by @copybara-service in #222
• Fix examples in the docs & linter warnings. by @copybara-service in #235
• Replace uses of jax.lib.xla_extension.Device with jax.Device. by @copybara-service in #237

Full Changelog: v0.1.5...v0.1.6

Chex 0.1.5

What's Changed

• Add support for value assertions in jitted functions. by @copybara-service in #178
• [JAX] Avoid private implementation detail _ScalarMeta. by @copybara-service in #180
• [JAX] Avoid implicit references to jax._src. by @copybara-service in #181
• Release v0.1.15 by @copybara-service in #184

Full Changelog: v0.1.4...v0.1.5

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

Chex 0.1.3

What's Changed

• Slight helping clarification to clear_trace_counter. by @lucasb-eyer in #148
• Add new JAX-specific pytypes to chex pytypes. by @copybara-service in #153
• Remove chex.{C,G,T}puDevice in favour of chex.Device. by @copybara-service in #154

New Contributors

• @lucasb-eyer made their first contribution in #148

Full Changelog: v0.1.2...v0.1.3

Chex 0.1.2

What's Changed

• Support JAX parallel operations in chex.fake_pmap contexts by @copybara-service in #142
• Remove references to jax.numpy.lax_numpy. by @copybara-service in #150

Full Changelog: v0.1.1...v0.1.2

Chex 0.1.1

What's Changed

• Move dataclass registration to init so that it's invoked after deserialization. by @copybara-service in #111
• Add pytype for jax array's dtypes. by @copybara-service in #112
• Fix dataclass registration on deserialization. by @copybara-service in #114
• Fix restrict_backends after jax.xla.backend_compile was moved by @copybara-service in #116
• Refactor asserts.py and warn users not to rely on asserts_internal's functionality. by @copybara-service in #117
• Set up ReadTheDoc pages and add a few examples. by @copybara-service in #118
• Include Sphinx builds into CI tests. by @copybara-service in #119
• Adds internal functionality by @copybara-service in #122
• Update Chex citation. by @copybara-service in #125
• Refactor assertions in preparation for including them into the RTD docs. by @copybara-service in #126
• Add asserts, variants, and pytypes modules to the RTD docs. by @copybara-service in #127
• Fix references to collections.abc.Mappable -> collections.abc.Mapping in docs and comments. collections.abc.Mappable does not exist. by @copybara-service in #129
• Document the rational behing the mappability of chex.dataclasses. by @copybara-service in #130
• Add 3 new tree assertions: by @copybara-service in #131
• Add assert_tree_is_sharded for asserting that a tree is sharded across the specified devices. by @copybara-service in #132
• Add PyTreeDef to pytypes. by @copybara-service in #134
• Disallow ShardedDeviceArrays in assert_tree_is_on_host and assert_tree_is_on_device. by @copybara-service in #133
• Bump ipython from 7.16.1 to 7.16.3 in /requirements by @dependabot in #135
• Remove the old venv directory before testing the package. by @copybara-service in #138
• Refactor asserts.py in preparation for experimental device assertions. by @copybara-service in #137
• Fix minor typo in docs. by @copybara-service in #139
• Improve exception message for assert_tree_shape_prefix. by @copybara-service in #143
• Release v0.1.1 by @copybara-service in #146

Full Changelog: v0.1.0...v0.1.1

Chex 0.1.0

Support for Python 3.6 has been dropped as per JAX deprecation policy. Please upgrade to a supported Python version.

Chex 0.0.9

It is the latest version compatible with Python 3.6. See google-deepmind/optax#222 for more details.

Changes since 0.0.8:

• Use rtol=1e-6 in asserts.assert_tree_close;
• Added asserts.assert_trees_all_equal;
• Removed restricted_inheritance option from Chex dataclasses;
• Added dims= option to assert_equal_shape, to check a subset of dims;
• Added test.sh for launching CI tests on a local machine;
• Added support for default exception messages and types to assertions;
• Added support for jnp.bfloat16 to asserts.assert_trees_all_close();
• Added support for static_argnames to variants.with_jit;
• Added a restrict_backends module for constraining the set of backends that a region of code can use;
• Added asserts.assert_trees_all_equal_dtypes assertion;
• Exposed asserts.assert_tree_shape_suffix to the public API;
• Added asserts.assert_tree_shape_suffix to check whether arrays share the same suffix.

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

Chex 0.0.7

Changelog

Full Changelog

Closed issues:

• [REQ] Conda recipe #37

Merged pull requests:

• Release Chex 0.0.7 #50 (copybara-service[bot])
• Add is_divisible assertion. #49 (copybara-service[bot])
• Create py.typed #46 (graingert)
• Remove "# Lint as: python3" directives. #45 (copybara-service[bot])
• Pass arguments to jax.jit other than the function as kwargs instead of positional arguments. #43 (copybara-service[bot])
• [JAX] Use a hand-written Python wrapper for CompiledFunction. #42 (copybara-service[bot])
• Ensure LICENSE files are an exact copy of Apache LICENSE-2.0.txt [0]. #41 (copybara-service[bot])
• Fix @assert_max_traces usage checks. #40 (copybara-service[bot])
• Fix tree path formatting. #39 (copybara-service[bot])

* This Changelog was automatically generated by github_changelog_generator