Patrick Kidger
e05985df2b
Document IPython extension and version nump
2023-09-20 11:33:19 -07:00
Roma Knyaz
f454cb797c
Make jaxtyping an IPython extension
2023-09-20 10:34:32 -07:00
Patrick Kidger
d2785baced
Better suppotr for incomplete JAX installations. Supersedes #105 .
2023-09-14 20:13:06 -07:00
Patrick Kidger
c80c1264d3
Error message now mentions both array type and shape requirements. Supersedes #107 .
2023-09-14 20:09:48 -07:00
Patrick Kidger
e308695293
Updated to support both new and old style JAX PRNG keys, as they are going to co-exist simultaneously. See https://github.com/google/jax/pull/17297
2023-09-14 20:00:46 -07:00
Patrick Kidger
e347c480d5
Hash is now stable across python runtimes
2023-08-17 16:23:33 +01:00
Patrick Kidger
13e6870fb8
Compatibility with JAX changes to opaque dtypes
2023-07-26 09:33:32 -07:00
Patrick Kidger
4c90808401
jaxtyping no longer appears in tracebacks
2023-07-26 09:33:32 -07:00
Patrick Kidger
5a57456e15
document Levanter
2023-07-12 19:36:09 +01:00
Patrick Kidger
a6ab6c0d28
Have PRNGKeyArray appear correctly in downstream documentation
2023-06-29 18:39:22 -07:00
Patrick Kidger
83be9e9d16
Fixed jaxtyping doc generation
2023-06-25 12:00:16 -07:00
Patrick Kidger
d2aa9c1e8d
Merge branch 'main' of https://github.com/google/jaxtyping
2023-06-14 10:42:32 -07:00
Patrick Kidger
926dc53856
Have PyTreeDef appear correctly in docs
2023-06-14 10:42:14 -07:00
Patrick Kidger
edc34f14f8
Update ecosystem links.
2023-06-07 15:35:45 +01:00
Patrick Kidger
8fa15050bc
Update ecosystem links.
2023-06-07 15:35:14 +01:00
Patrick Kidger
356f5b7f7b
Build fixes
2023-06-01 11:06:02 -07:00
Patrick Kidger
1b9c9fab52
Bump to Py3.9
2023-06-01 10:56:00 -07:00
Patrick Kidger
066a5b058f
Made modules private.
2023-06-01 10:56:00 -07:00
Patrick Kidger
319d54abcf
Avoid __builtins__ getting added as a key
2023-06-01 10:56:00 -07:00
Patrick Kidger
6a64ef114e
Now provides PyTreeDef, and can detect PyTrees via issubclass(x, PyTree)
2023-06-01 10:56:00 -07:00
Patrick Kidger
10e1852b37
Fix favicon
2023-05-12 11:57:06 -07:00
Patrick Kidger
a19149d23d
Fixed pytest hook
2023-05-11 09:08:09 -07:00
Patrick Kidger
0c596ff373
Fix for non-JAX installations.
2023-05-10 17:16:42 -07:00
Patrick Kidger
7934d2afed
Static typing fixes
2023-05-10 16:35:45 -07:00
Patrick Kidger
849b15db3b
Added better docs; added PRNGKeys. Packaging with pyproject.toml.
...
Also added support for matching dtypes using regexes, and for nesting e.g. Shaped[Float[Array, "dim1 dim2"], "dim3"].
2023-05-10 15:27:25 -07:00
Patrick Kidger
46c7896c99
Transitive test now uses typeguard instead of beartype
2023-04-19 11:48:08 -07:00
Patrick Kidger
c92b0d0ab1
Some improvements ( #77 )
...
* Various improvements.
- Added support for functions in symbolic dimensions, e.g. "min(foo,bar)", which were previously disallowed due to the presence of a comma. (#51 )
- Added support for adding ignored names to dimensions, e.g. "cols=4". (#76 )
* Now works with Python 3.10 A | B union types.
2023-04-13 19:18:55 +01:00
Patrick Kidger
158b8b8f0c
Now works with torch.compile? ( #72 )
2023-04-13 18:53:14 +01:00
Patrick Kidger
9b6df18b83
Update FAQ to mention ruff
2023-03-19 22:21:21 +00:00
Patrick Kidger
f0b240df5f
Switched to ruff
2023-03-15 22:36:24 -07:00
Patrick Kidger
a4d27c7cc1
Fixed _Jaxtyped.__get__, e.g. swallowing abstractmethod decorations
2023-03-15 22:27:36 -07:00
Patrick Kidger
ee46c57e53
Version bump
2023-03-05 20:26:01 -08:00
Patrick Kidger
38be24f9c8
beartype+inheritance fix. Bool[int, '...'] now correctly raises an error.
2023-03-05 20:09:49 -08:00
Patrick Kidger
e718f00cc5
Fixed import hook hitting __pycache__ even when you change the choice of runtime type checker
2023-03-05 16:19:44 -08:00
Patrick Kidger
c232eeaa89
Fixed pytest plugin with new import hook typechecker syntax
2023-03-05 12:19:14 -08:00
Patrick Kidger
e03c1c329e
We now have Float[np.ndarray, ...] <: np.ndarray. Added basic torch tests. ( #68 )
...
This required quite a lot of refactoring! JAX supports virtual subclass registration (its metaclass is ABCMeta) but NumPy does not, so we have to actually subclass `np.ndarray`.
Simple stuff like __base__ hacking fails due to deallocator conflicts.
2023-03-04 17:29:04 +00:00
Patrick Kidger
fef81cf0a0
The import hook now supports BeartypeConf/BeartypeStrategy
2023-03-03 10:34:03 -08:00
Patrick Kidger
bf241b4e27
We now have e.g. Float[Array, ""] <: Array.
2023-03-03 10:32:26 -08:00
Patrick Kidger
5600a1aac8
Fixed cloudpickle breaking, mark 2
2023-03-02 17:37:53 -08:00
Patrick Kidger
2b339715f9
Fixed cloudpickle breaking
2023-03-02 12:35:38 -08:00
Zac Cranko
8c86958b77
Add TypeAlias decoration to PyTree ( #66 )
...
Doing this silences a *whole heap* of Pyright warnings that all say "Illegal type annotation: variable not allowed unless it is a type alias"
2023-02-28 01:23:47 +00:00
Patrick Kidger
ffc56bf782
Edge case fix
2023-02-25 17:38:45 -08:00
Patrick Kidger
5c25da278a
Bump version
2023-02-25 17:03:25 -08:00
Patrick Kidger
e2f004afd4
Added support for jax.typing.ArrayLike; now works with PyTorch's bool
2023-02-25 17:01:27 -08:00
Patrick Kidger
81c56052e5
Fixes for some new failures. (Where did they come from?) ( #65 )
...
* Fixes for some new failures. (Where did they come from?)
* Fixed isort?
2023-02-16 10:08:57 -08:00
Patrick Kidger
d911ebb99c
Fix abstractmethods being ignored after @jaxtyped
2023-01-22 11:43:36 -08:00
Patrick Kidger
f30b7d1546
Update README.md
2023-01-20 07:44:45 -08:00
Brent Yi
4b3f834e12
Fix vanilla dataclasses ( #56 )
2023-01-15 10:52:18 +01:00
Patrick Kidger
59e8fb0d18
Hopefully fixed PyTree raising spurious errors. Bit mysterious that this worked before, really. I've tested this fix as best I can against the various static type checkers, but these are weird and varied enough that this might not be a perfect fix. If you see this and have issues, let me know. ( #54 )
2022-12-30 19:00:26 +00:00
Patrick Kidger
7dba3516c2
Fixed working with the new (unreleased) version of typeguard ( #53 )
2022-12-29 17:46:25 +00:00
Patrick Kidger