patrick-kidger/jaxtyping v0.2.14 on GitHub

Highlights

Silenced spurious pyright>=1.1.293 errors with jaxtyping.PyTree. (Thanks @ZacCranko in #66!)
Compatibility with Plum v2 (beartype/plum#73).

Features

Tweaked install_import_hook(..., typechecker=...) argument to also accept a string. In particular this means it can be used with beartype's new full checking; this can be enabled by passing typechecker="beartype.beartype(conf=beartype.BeartypeConf(strategy=beartype.BeartypeStrategy.On))".
It is now the case that e.g. issubclass(Float[array_type, ...], array_type). (This is what brings compatibility with Plum v2.)
NumPy and PyTorch compatibility is now tested.

Bugfixes

Statements like Bool[int, "..."] should now raise an error.
Applying @jaxtyping fo functions with fn in their __dict__ should now work.
Subclassing dataclasses whilst using install_import_hook(..., typechecker="beartype.beartype") should no longer raise a spurious error.
Changing the value of install_import_hook(..., typechecker=...) will no longer wrongly hit the same __pycache__. (Before this the change in value was ignored.)
Can now serialise jaxtyping types using cloudpickle.

New Contributors

Full Changelog: v0.2.13...v0.2.14