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
2b1be5eb0a
Update README.md
2022-12-07 17:22:57 -08:00
Brent Yi
7b3d9a2e9a
Explicitly export names in jaxtyping.* ( #49 )
...
* Explicitly export names to make pyright happy
* Bump jax and jaxtyping versions
* Add note on `jaxtyping` names
* Remove __all__ from `array_types.py`
* Appease flake8
* Reduce import redundancy
* Fix capitalization
2022-12-07 17:21:33 -08:00
Patrick Kidger
29654e7087
JAX is no longer a hard dependency (to support e.g. PyTorch) ( #50 )
2022-12-07 10:48:47 -08:00
Patrick Kidger
8fbf7bf3a5
added link to eqxvision
2022-12-05 11:37:32 -08:00
Patrick Kidger
a220df9964
The import hook now decorates dataclass __init__ methods ( #48 )
2022-11-16 13:38:04 -08:00
Kevin P Murphy
784aa78f7c
update jaxtyped decorator ( #44 )
...
* update jaxtyped decorator
* add newline character to pacify flake
* add precomit hooks
* add missing return statement
* replace typing_extensions>=4.2.0 with typing_extensions
* pin version range for typing_extensions
* set min version of typing-extensions but not max
* bump version number to 0.2.8
2022-11-13 22:25:28 -08:00
Patrick Kidger
3f877c0dbb
Update array_types.py ( #41 )
2022-11-08 22:52:17 -08:00
Patrick Kidger
607f3c66b5
Silenced warning ( #40 )
2022-10-28 15:19:52 -07:00
Peter Roelants
d3651ca70e
NamedTuple example ( #36 )
2022-10-03 07:40:51 -07:00
Patrick Kidger
d246e21281
Better import hook ( #35 )
2022-09-25 23:28:40 -07:00
Patrick Kidger
165065756f
Static type-checking fix ( #34 )
2022-09-24 18:27:22 -07:00
ebrevdo
dcd73e3431
Add support for e.g. jaxtyping.Float[Union[...], ...] in py3.8 ( #31 )
...
* Add support for e.g. jaxtyping.Float[Union[...], ...] in py3.8
Turns out that python3.8, Union lacks the __name__ attribute. Use repr()
in these cases.
* Fix linter.
* Remove implicit cast to bool in favor of try/except.
2022-09-22 13:25:29 -07:00
Patrick Kidger
f175c7f315
Fixed py.type not being packaged ( #30 )
2022-09-22 12:16:25 -07:00
Patrick Kidger
da8300ec6c
tidyness tweak ( #28 )
2022-09-20 15:16:52 -07:00
Patrick Kidger
39439c2790
More fixes ( #27 )
...
* Edge-case doc fixes for parameterising types with PyTrees or AbstractDtypes
* Fixes for threading
* version bump
2022-09-20 15:12:21 -07:00
Patrick Kidger
6202dcc639
doc fix ( #26 )
2022-09-19 23:30:29 -07:00
Patrick Kidger
c2e9d913d5
Fixed jaxtyped breaking descriptors. Fixed long module names. ( #25 )
2022-09-19 22:40:25 -07:00
Patrick Kidger
62ddcc25b5
Threading fix ( #24 )
2022-09-16 08:04:28 -07:00
Patrick Kidger
a89ebe356b
Merge pull request #22 from google/v021
...
Bump version
2022-09-15 08:19:08 -07:00
Patrick Kidger
98133f5e1e
Bump version
2022-09-15 16:15:28 +01:00
Patrick Kidger
2c7dbbd593
Merge pull request #21 from google/pytyped
...
Create `py.typed`
2022-09-15 08:15:03 -07:00
Patrick Kidger