jaxtyping v0.3.10

Bugfixes

• add u32[4] as a possible PRNGKeyArray type (Thanks @jondeaton! #388)
• Make variadic objects (e.g. Float[Array, "..."]) survive cloudpickle round-trip. (Thanks @hmgaudecker! #391)

Infrastructure

• Use dependency groups, uv, and local pre-commit hooks (#386)
• Add pyrefly as a type checker in pre-commit and dev dependencies (Thanks @lolpack! #387)

New Contributors

• @patrick-kidger-bot made their first contribution in #385
• @lolpack made their first contribution in #387
• @jondeaton made their first contribution in #388
• @hmgaudecker made their first contribution in #391

Full Changelog: v0.3.9...v0.3.10

jaxtyping v0.3.9

Bugfix: typechecking of pytrees of |-style unions, e.g. isinstance(..., PyTree[int | bool]), will no longer silently always pass. #381

Full Changelog: v0.3.8...v0.3.9

jaxtyping v0.3.8

Another bugfix release for numpy.typing.ArrayLike on numpy 2.4.0, I think this time only needed Python 3.13+. #380

Full Changelog: v0.3.7...v0.3.8

jaxtyping v0.3.7

Bugfix: SomeDtype[Union[numpy.typing.ArrayLike, ...], ...] will no longer crash in numpy 2.4.0. (#374)

Full Changelog: v0.3.6...v0.3.7

jaxtyping v0.3.6

• Bugfix: @typing.no_type_check is now respected on dataclasses. (Thanks @jeertmans! #370)
• Bugfix: the pytest plugin is now exected before pytest conftest.py is ran, making it usable even if the package under test is imported inside conftest.py. (Thanks @jeertmans! #371)
• Bugfix: the pytest plugin supports typecheckers that call multi-argument constructors. (Thanks @jeertmans! #369)

Full Changelog: v0.3.5...v0.3.6

jaxtyping v0.3.5

• Feature: support TypeAliasTypes as array types. For example:

  type Foo = np.ndarray | int
  Float[Foo, ""]

  In particular, this re-enables support for numpy.typing.ArrayLike, which recently switched from being a union to a type alias.

• Docs: big update to emphasize the way in which we are framework-agnostic.

Full Changelog: v0.3.4...v0.3.5

jaxtyping v0.3.4

• Add fp4e2m1fn type (Thanks @vincentlo-a! #353)
• No longer crashing if numpy not available. (Thanks @charlesbmi! #361)
• No longer crashing when building documentation using either either pytkdoc_tweaks or hippogriffe (=my internal doc libraries that probably no-one else uses 😁) and JAX is not available (#362)

New Contributors

• @vincentlo-a made their first contribution in #353
• @charlesbmi made their first contribution in #361

Full Changelog: v0.3.3...v0.3.4

jaxtyping v0.3.3

A collection of esoteric fixes, mostly due to versions of other libraries:

• Fix for partially-stringified annotations resulting in spurious errors. (If we can't resolve them at decoration time then they just become an Any instead.)
• Fix for Dtype[numpy.typing.ArrayLike, ...] not accepting bool/int/float in NumPy 2.3.3-ish.
• Fix for Foo | Bar annotations with typeguard==2.13.3 producing false negatives. (Fixes #349.)
• Fix for jaxtyping.ArrayLike not including jax._src.literals.LiteralArray under jax==0.7.2.

New Contributors

• @nstarman made their first contribution in #345

Full Changelog: v0.3.2...v0.3.3

jaxtyping v0.3.2

Autogenerated release notes as follows:

What's Changed

• Add jaxtyped ContextManager overload annotation. by @asford in #316
• Now robust to all Exceptions during type hint resolution, not just NameErrors. by @patrick-kidger in #317

Full Changelog: v0.3.1...v0.3.2

jaxtyping v0.3.1

• Minor bugfix: dataclass __init__ methods are no longer double-wrapped in jaxtyping.jaxtyped when using the import hook. (#313)

Full Changelog: v0.3.0...v0.3.1