Bump jaxtyping from 0.2.21 to 0.2.31 in /training

[dependabot]

Bumps jaxtyping from 0.2.21 to 0.2.31.

Release notes

Sourced from jaxtyping's releases.

jaxtyping v0.2.31

• Now duck-type on array shapes and dtypes, so you can use jaxtyping for your custom arraylike objects:

  class FooDtype(jaxtyping.AbstractDtype):
  	dtypes = ["foo"]
  class MyArray:
  @​property
  def dtype(self):
  return "foo"
  @property
  def shape(self):
  	return (3, 1, 4)
  
  def f(x: FooDtype[MyArray, "3 1 4"]): ...

• Improved compatibility when typeguard warns that you're typechecking a function without annotations: it will no longer mention the jaxtyping-internal check_params function and will instead mention the name of the function that is missing annotations.

• Improved the error message when typechecking fails, to state the full some_module.SomeClass.some_method rather than just some_method.

• Fixed a JAX deprecation warning for jax.tree_map. (Thanks @​groszewn!)

New Contributors

• @​groszewn made their first contribution in patrick-kidger/jaxtyping#218

Full Changelog: patrick-kidger/jaxtyping@v0.2.30...v0.2.31

jaxtyping v0.2.30

• Now reporting the correct source code line numbers when using the import hook. Makes debuggers useful again! #214
• Now supports numpy structured dtypes (Thanks @​alexfanqi! #211)
• Now respecting typing.no_type_check. #216

New Contributors

• @​alexfanqi made their first contribution in patrick-kidger/jaxtyping#211

Full Changelog: patrick-kidger/jaxtyping@v0.2.29...v0.2.30

jaxtyping v0.2.29

• Crash fix for when jax is available but jaxlib is not. (Thanks @​ar0ck! #191)
• Crash fix when used alongside old TensorFlow versions that don't support tensor.ndim (Thanks @​dziulek! #193)
• Crash fix when using a default argument as a symbolic dimension size. (Thanks @​jaraujo98! #208)
• Improved import times by defining the IPython magic lazily. (Thanks @​superbobry! #201)
• The import hook will now typecheck functions that do not have any annotations in the arguments or return value. This is useful for those that do manual isinstance checks in the body of teh function. (Thanks @​nimashoghi! #205)
• Dropped the dependency on numpy. This makes it possible to just use jaxtyping+typeguard as the one-stop-shop for all runtime typechecking, even when you're not using arrays. Obviously that's a little unusual -- not really the main focus of jaxtyping -- but helps when wanting a single choice of runtime type checker across an entire codebase, only parts of which may use arrays. (#212)

New Contributors

• @​ar0ck made their first contribution in patrick-kidger/jaxtyping#191

... (truncated)

Commits

• b0bbff9 Added support for custom arrays.
• 099b189 slightly better compatibility again
• 4716fc1 typeguard warnings, on not having any annotations, now name the correct funct...
• 40f494b Now specifying module and qualname in error messages, for easier debugging.
• 883c604 Added some __tracebackhide__s
• 4428513 Update deprecate method usage
• 10b5722 bump version
• 6ae28a4 Now respecting typing.no_type_check decorator.
• 619f22e Fixes for docs and no-numpy dependencies.
• ad95c10 Add support for numpy struct dtype
• 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 #69.