wassname/jaxtyping
1
0
Fork 0
mirror of https://github.com/wassname/jaxtyping.git synced 2026-09-12 12:32:19 +08:00
Code Issues Packages Projects Releases Wiki Activity

Adding a test for generator support (#171)

Browse Source 
* Add a test for generators

* Remove output annotations from decorators

Also guarded torch imports for better compatibility with
requirements.txt

* Add flag to the main meta class to skip the typecheck

* Return to the old solution

* Make async tests work

* Minor adjustments/fixing typos

* Correct Python path for new tests

* Remove some jax-dependent code

* Implement equality for MetaArrays

* Make all Dim variations frozen dataclasses

* Shorten AbstractArray methods

* Final touches

* Removing get_origin use

* Update tests with @jaxtyp
This commit is contained in:
Roman Knyazhitskiy
2024-02-25 12:07:01 +00:00
committed by Patrick Kidger
parent 17ea4b13eb
commit 172b83b4fc
8 changed files with 246 additions and 43 deletions
Download Patch File Download Diff File Expand all files Collapse all files
test/requirements.txt
+1
View File
@@ -4,5 +4,6 @@ equinox
IPython
jaxlib
pytest
pytest-asyncio
tensorflow
typeguard<3