Bump jaxtyping from 0.2.21 to 0.2.34 in /training

[dependabot]

Bumps jaxtyping from 0.2.21 to 0.2.34.

Release notes

Sourced from jaxtyping's releases.

jaxtyping v0.2.34

• Compatibility with ray -- this fixes crashes with the error message has no attribute 'index_variadic' (#198, #237)
• Compatibility with Python 3.12: fixed deprecation warnings about ast.Str. (#236, thanks @​phinate!)
• Error message improvements + doc improvements (#233, #240, thanks @​padix-key and @​jjyyxx!)

New Contributors

• @​padix-key made their first contribution in patrick-kidger/jaxtyping#233
• @​phinate made their first contribution in patrick-kidger/jaxtyping#236
• @​jjyyxx made their first contribution in patrick-kidger/jaxtyping#240

Full Changelog: patrick-kidger/jaxtyping@v0.2.33...v0.2.34

jaxtyping v0.2.33

• Compatibility with Python 3.10 when using Any as the array type.
• Compatibility with generic array types.
• Array typevars now respect __constraints__

Full Changelog: patrick-kidger/jaxtyping@v0.2.32...v0.2.33

jaxtyping v0.2.32

• The array type can now be either Any or a TypeVar. In both cases this means that anything is allowed at runtime. As usual, static type checkers will only look at the array part of an annotation, so that an annotation of the form Float[T, "foo bar"] (where T = TypeVar("T")) will be treated as just T by static type checkers. This allows for expressing array-type-polymorphism with static typechecking. Here's an example:

  import numpy as np
  import torch
  from typing import TypeVar
  TensorLike = TypeVar("TensorLike", np.ndarray, torch.Tensor)
  def stack_scalars(x: Float[TensorLike, ""], y: Float[TensorLike, ""]) -> Float[TensorLike, "2"]:
  if isinstance(x, np.ndarray) and isinstance(y, np.ndarray):
  return np.stack([x, y])
  elif isinstance(x, torch.Tensor) and isinstance(y, torch.Tensor):
  return torch.stack([x, y])
  else:
  raise ValueError("Invalid array types!")

• Fixed a bug in which the very first argument to a function was erroneously reported as the one at fault for a typechecking error. This bug occurred when using default arguments.

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

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:

... (truncated)

Commits

• f4ca4c9 version bump
• ec71bf0 Fix broken link
• b3a13eb Remove polyfill
• 09235dd Remove redundant colon in install_import_hook docstring
• f9e44cf Compatibility with ray.
• 317cc9e update CONTRIBUTING.md
• 2ae053b replace ast.Str with ast.Constant to avoid deprecation warnings in python 3.12
• b877cf8 Fix misleading error message
• c2f19db Compatibility with Python 3.10 + with generic array types
• f7c57c3 Version bump
• 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 #72.