Add stricter type checking and expected matrix sizes#42

Merged

gpleiss merged 75 commits intocornellius-gp:mainfrom

corwinjoy:jaxtyping

May 2, 2023

Commits on Nov 9, 2022

Copy files from typing to this branch
Corwin Joy
committed

Commits on Nov 10, 2022

Fix isort config to respect black
Corwin Joy
committed
Add jax typing examples
Corwin Joy
committed

Commits on Nov 11, 2022

Setup tests to check runtime signatures and begin working through failed test cases.
Corwin Joy
committed
Continue to fix signatures
Corwin Joy
committed

Commits on Dec 2, 2022

Got runthrough of test cases working after upgrading to latest pytorch
Corwin Joy
committed
Finish signatures for _linear_operator.py
Corwin Joy
committed
Do signatures for added_diag_linear_operator.py
Corwin Joy
committed
Do signatures for added_diag_linear_operator.py p2
Corwin Joy
committed

Commits on Dec 3, 2022

Upgrade signatures and expand scope of runtime tests.
Corwin Joy
committed

Commits on Dec 6, 2022

Continue fixing function signature to pass runtime tests.
Corwin Joy
committed
Upgraded signatures and tests for add_diag_linear_operator.py
Corwin Joy
committed

Commits on Dec 7, 2022

Upgraded signatures and tests for batch_repeat_linear_operator.py
Corwin Joy
committed
Upgraded signatures and tests for block_diag_linear_operator.py
Corwin Joy
committed
Upgraded signatures and tests for block_interleaved_linear_operator.py
Corwin Joy
committed
Upgraded signatures and tests for block_linear_operator.py
Corwin Joy
committed
Upgraded signatures and tests for cat_linear_operator.py.py
Corwin Joy
committed

Commits on Dec 8, 2022

Upgraded signatures and tests for chol_linear_operator.py
Corwin Joy
committed
Upgraded signatures and tests for constant_mul_linear_operator.py
Corwin Joy
committed
Upgraded signatures and tests for dense_linear_operator.py
Corwin Joy
committed

Commits on Dec 9, 2022

Upgraded signatures and tests for identity_linear_operator.py
Corwin Joy
committed
Upgraded signatures and tests for interpolated_linear_operator.py
Corwin Joy
committed

Commits on Dec 10, 2022

Commits on Dec 14, 2022

Upgraded signatures and tests for low_rank_root_added_diag_linear_operator.py
Corwin Joy
committed
Upgraded signatures and tests for low_rank_root_linear_operator.py
Corwin Joy
committed
Upgraded signatures and tests for matmul_linear_operator.py
Corwin Joy
committed
Upgraded signatures and tests for mul_linear_operator.py
Corwin Joy
committed
Upgraded signatures and tests for psd_sum_linear_operator.py
Corwin Joy
committed
Upgraded signatures and tests for root_linear_operator.py
Corwin Joy
committed

Commits on Dec 15, 2022

Upgraded signatures and tests for sum_batch_linear_operator.py
Corwin Joy
committed
Upgraded signatures and tests for sum_kronecker_linear_operator.py
Corwin Joy
committed
Upgraded signatures and tests for sum_linear_operator.py
Corwin Joy
committed
Upgraded signatures and tests for toeplitz_linear_operator.py
Corwin Joy
committed
Upgraded signatures and tests for triangular_linear_operator.py
Corwin Joy
committed
Upgraded signatures and tests for zero_linear_operator.py
Corwin Joy
committed

Commits on Dec 16, 2022

Upgraded signatures for _linear_operator.py
Corwin Joy
committed
Merged updates to main branch
Corwin Joy
committed
Add jaxtyping to the requirements in setup.py
Corwin Joy
committed

Commits on Dec 17, 2022

Commits on Dec 30, 2022

Commits on Jan 24, 2023

Update propagate_type_hints.py

corwinjoy
and
Balandat
authored
Remove reference to EllipsisType to support older versions of Python
Corwin Joy
committed

Commits on Mar 19, 2023

Respond to review comments from Geoff Pleiss. Make the jaxtyping optional and clean up a couple pieces of dead code.
Corwin Joy
committed
Minor cleanup to propagate_type_hints.py.
Corwin Joy
committed
Merge branch 'main' into jaxtyping
Corwin Joy
committed
Remove unwanted additions to _linear_operator.py vs what is in main.
Corwin Joy
committed

Commits on Mar 21, 2023

Commits on Mar 22, 2023

Setup pytest typecheck configuration and run it all the way through.
Corwin Joy
committed
Setup pytest typecheck configuration and run it all the way through.
Corwin Joy
committed

Commits on Apr 19, 2023

Merge branch 'main' into jaxtyping
gpleiss
committed

Commits on Apr 20, 2023

Commits on May 2, 2023