Bump jaxtyping from 0.2.21 to 0.2.28 in /training

[dependabot]

Bumps jaxtyping from 0.2.21 to 0.2.28.

Release notes

Sourced from jaxtyping's releases.

jaxtyping v0.2.28

Autogenerated release notes as follows:

What's Changed

• Fixes #188. by @​patrick-kidger in patrick-kidger/jaxtyping#190

Full Changelog: patrick-kidger/jaxtyping@v0.2.27...v0.2.28

jaxtyping v0.2.27

Quick bugfix release:

• Fixed some isinstance checks against variadics crashing (although this was when it was about to return False anyway). (Thanks @​asford! #186)
• Fixed docs for downstream libraries (Equinox, ...) not generating correctl (#182)

New Contributors

• @​asford made their first contribution in patrick-kidger/jaxtyping#186

Full Changelog: patrick-kidger/jaxtyping@v0.2.26...v0.2.27

jaxtyping v0.2.26

Features

• Added jaxtyping.print_bindings to manually inspect the values of each axis, whilst inside a function.
• Added support for jaxtyping.{Int4, UInt4}. (#174, thanks @​jianlijianli!)

Bugfixes

• We no longer import JAX at all, even if it is present. This ensures compatibility when using jaxtyping+PyTorch alongside an old JAX installation. (All JAX re-exports, like jaxtyping.Array = jax.Array, are looked up dynamically rather than import time.) (#178)
• We no longer raise false postiives when @jaxtyped-ing generators (with yield statements). (#91, #171, thanks @​knyazer!)

Internals

• Added support for beartype's pseudostandard __instancecheck_str__ method. Instead of isinstance(x, Float[Array, "foo"]), then one can now call Float[Array, "foo"].__instancecheck_str__(x), which will return either an empty string (success) or an error message describing why the check failed (wrong shape, wrong dtype, ...). In practice this feature probably isn't super usable right now; we'll need to wait until we've later done a better job ensuring compatibility between the jaxtyping import hooks and the beartype import hooks.

Docs

• Fixes by @​jeertmans (#154) and @​afrozenator (#170) -- thank you!

New Contributors

• @​jeertmans made their first contribution in patrick-kidger/jaxtyping#154
• @​afrozenator made their first contribution in patrick-kidger/jaxtyping#170

Full Changelog: patrick-kidger/jaxtyping@v0.2.25...v0.2.26

jaxtyping v0.2.25

This release is primarily a usability release, designed to help ensure the library is being used correctly.

• The error messages from a failed typecheck have been improved, to explicitly highlight more information about which argument was wrong. :)
• If the jaxtyping.jaxtyped(typechecker=...) argument is not passed, then a warning will be displayed. In practice, this will trigger:

... (truncated)

Commits

• 102e499 Fixes #188.
• f708d17 Quick fix for docs not generating correctly since the last jaxtyping release
• 5e2518c Fix _check_shape str formatting for variadics
• 1b3173a Bump version
• 172b83b Adding a test for generator support (#171)
• 17ea4b1 No longer imports JAX at all! This is done dynamically when required. See #178
• 9beb5f2 Add int4/uint4 support in jaxtyping. (#174)
• 1d4d402 Added support for beartype 0.17.0's instancecheck_str.
• 28ad527 Moved print_bindings into storage.py
• d7fd59a Added print_bindings.
• Additional commits viewable in compare view

Dependabot will resolve any conflicts with this PR as long as you don't alter it yourself. You can also trigger a rebase manually by commenting @dependabot rebase.

---

Dependabot commands and options

You can trigger Dependabot actions by commenting on this PR:

• @dependabot rebase will rebase this PR
• @dependabot recreate will recreate this PR, overwriting any edits that have been made to it
• @dependabot merge will merge this PR after your CI passes on it
• @dependabot squash and merge will squash and merge this PR after your CI passes on it
• @dependabot cancel merge will cancel a previously requested merge and block automerging
• @dependabot reopen will reopen this PR if it is closed
• @dependabot close will close this PR and stop Dependabot recreating it. You can achieve the same result by closing it manually
• @dependabot show <dependency name> ignore conditions will show all of the ignore conditions of the specified dependency
• @dependabot ignore this major version will close this PR and stop Dependabot creating any more for this major version (unless you reopen the PR or upgrade to it yourself)
• @dependabot ignore this minor version will close this PR and stop Dependabot creating any more for this minor version (unless you reopen the PR or upgrade to it yourself)
• @dependabot ignore this dependency will close this PR and stop Dependabot creating any more for this dependency (unless you reopen the PR or upgrade to it yourself)

[dependabot]

Superseded by #58.