Roman Knyazhitskiy 172b83b4fc Adding a test for generator support (#171) ...

* 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

2024-02-25 12:07:01 +00:00

..

__init__.py

Make test/helpers.py a relative import

2022-07-12 10:03:19 +01:00

conftest.py

Added environment config flags.

2023-12-05 19:06:00 -08:00

helpers.py

Fixed working with the new (unreleased) version of typeguard (#53)

2022-12-29 17:46:25 +00:00

import_hook_tester.py

dataclasses now have fields checked, not __init__.

2023-10-09 21:58:14 -07:00

requirements.txt

Adding a test for generator support (#171)

2024-02-25 12:07:01 +00:00

test_all_importable.py

No longer imports JAX at all! This is done dynamically when required. See #178

2024-02-25 12:07:01 +00:00

test_array.py

Adding a test for generator support (#171)

2024-02-25 12:07:01 +00:00

test_decorator.py

Added print_bindings.

2024-02-25 12:07:01 +00:00

test_equals.py

Adding a test for generator support (#171)

2024-02-25 12:07:01 +00:00

test_generators.py

Adding a test for generator support (#171)

2024-02-25 12:07:01 +00:00

test_import_hook.py

Fixes for g3

2023-10-23 22:00:51 -07:00

test_ipython_extension.py

Better handling of user-defined typechecker

2023-10-09 10:21:07 -07:00

test_messages.py

Improved error messages a little bit, in particular to highlight individual problematic arguments.

2023-12-08 10:17:57 -08:00

test_no_jax_dependency.py

Adding a test for generator support (#171)

2024-02-25 12:07:01 +00:00

test_pytree.py

Simplified internals by removing jaxtyping_raise; jaxtyping_malformed.

2023-12-05 19:06:00 -08:00

test_serialisation.py

Adding a test for generator support (#171)

2024-02-25 12:07:01 +00:00

test_tf_dtype.py

Should be more robust to jax/numpy/tensorflow version changes

2023-09-25 11:07:36 -07:00

test_threading.py

Added environment config flags.

2023-12-05 19:06:00 -08:00