Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Introduce beartype & fix types #230

Merged
merged 91 commits into from
Apr 27, 2023
Merged

Introduce beartype & fix types #230

merged 91 commits into from
Apr 27, 2023

Commits on Apr 13, 2023

  1. add beartype dependency

    @st--
    st-- committed Apr 13, 2023
    Configuration menu
    Copy the full SHA
    15b808f View commit details
    Browse the repository at this point in the history

  2. from typing import -> from beartype.typing import

    @st--
    st-- committed Apr 13, 2023
    Configuration menu
    Copy the full SHA
    0eaec30 View commit details
    Browse the repository at this point in the history

  3. jaxtyping import_hook for @jaxtyped @beartype everywhere

    @st--
    st-- committed Apr 13, 2023
    Configuration menu
    Copy the full SHA
    781bc38 View commit details
    Browse the repository at this point in the history

  4. fix Type[] of class-as-argument

    @st--
    st-- committed Apr 13, 2023
    Configuration menu
    Copy the full SHA
    67e315f View commit details
    Browse the repository at this point in the history

  5. fix KeyArray type hint (should probably move into jaxutils though)

    @st--
    st-- committed Apr 13, 2023
    Configuration menu
    Copy the full SHA
    e77732b View commit details
    Browse the repository at this point in the history

  6. fix return value of slice_input when active_dims is None

    @st--
    st-- committed Apr 13, 2023
    Configuration menu
    Copy the full SHA
    5175fa7 View commit details
    Browse the repository at this point in the history

  7. fix return value of squared_distance

    @st--
    st-- committed Apr 13, 2023
    Configuration menu
    Copy the full SHA
    c548d73 View commit details
    Browse the repository at this point in the history

  8. fix return type of recursive_bijectors

    @st--
    st-- committed Apr 13, 2023
    Configuration menu
    Copy the full SHA
    079936f View commit details
    Browse the repository at this point in the history

  9. fix slice_input type annotations

    @st--
    st-- committed Apr 13, 2023
    Configuration menu
    Copy the full SHA
    1005212 View commit details
    Browse the repository at this point in the history

  10. new KernelCallable type to fix kernel_fn annotations

    @st--
    st-- committed Apr 13, 2023
    Configuration menu
    Copy the full SHA
    0c57a4e View commit details
    Browse the repository at this point in the history

  11. fix kernel __call__ annotation

    @st--
    st-- committed Apr 13, 2023
    Configuration menu
    Copy the full SHA
    777b042 View commit details
    Browse the repository at this point in the history

  12. fix KeyArray type hint

    @st--
    st-- committed Apr 13, 2023
    Configuration menu
    Copy the full SHA
    df88c69 View commit details
    Browse the repository at this point in the history

  13. beartype does not like forward references; replaced with string types

    @st--
    st-- committed Apr 13, 2023
    Configuration menu
    Copy the full SHA
    711fcac View commit details
    Browse the repository at this point in the history

  14. linops other type hint fixes

    @st--
    st-- committed Apr 13, 2023
    Configuration menu
    Copy the full SHA
    1c95fa5 View commit details
    Browse the repository at this point in the history

  15. fix KeyArray

    @st--
    st-- committed Apr 13, 2023
    Configuration menu
    Copy the full SHA
    1b915cf View commit details
    Browse the repository at this point in the history

  16. abstractions.py some type fixes

    @st--
    st-- committed Apr 13, 2023
    Configuration menu
    Copy the full SHA
    c3e48c1 View commit details
    Browse the repository at this point in the history

  17. fix GaussianDistribution.log_prob return type

    @st--
    st-- committed Apr 13, 2023
    Configuration menu
    Copy the full SHA
    4cc9db1 View commit details
    Browse the repository at this point in the history

  18. fix depreciations & warnings

    @st--
    st-- committed Apr 13, 2023
    Configuration menu
    Copy the full SHA
    4c4a572 View commit details
    Browse the repository at this point in the history

  19. Merge branch 'st/fix_depreciations' into st/beartype

    @st--
    st-- committed Apr 13, 2023
    Configuration menu
    Copy the full SHA
    c512195 View commit details
    Browse the repository at this point in the history

  20. fix scalar array types

    @st--
    st-- committed Apr 13, 2023
    Configuration menu
    Copy the full SHA
    9516fdf View commit details
    Browse the repository at this point in the history

  21. introduce ScalarBool, ScalarInt for jitted calls in abstractions

    @st--
    st-- committed Apr 13, 2023
    Configuration menu
    Copy the full SHA
    2d45258 View commit details
    Browse the repository at this point in the history

  22. relax LinearOperator's solve() types (can be both matrix or vector), … 

    …not ideal :S
    @st--
    st-- committed Apr 13, 2023
    Configuration menu
    Copy the full SHA
    ecfde64 View commit details
    Browse the repository at this point in the history

  23. remove _stop_grad type hints, not sure what they should be

    @st--
    st-- committed Apr 13, 2023
    Configuration menu
    Copy the full SHA
    066a5d7 View commit details
    Browse the repository at this point in the history

  24. found some more

    @st--
    st-- committed Apr 13, 2023
    Configuration menu
    Copy the full SHA
    ff68725 View commit details
    Browse the repository at this point in the history

  25. Merge branch 'st/fix_depreciations' into st/beartype

    @st--
    st-- committed Apr 13, 2023
    Configuration menu
    Copy the full SHA
    4716773 View commit details
    Browse the repository at this point in the history

  26. float -> ScalarFloat fixes

    @st--
    st-- committed Apr 13, 2023
    Configuration menu
    Copy the full SHA
    5c90755 View commit details
    Browse the repository at this point in the history

  27. linops log_det type fixes

    @st--
    st-- committed Apr 13, 2023
    Configuration menu
    Copy the full SHA
    c692568 View commit details
    Browse the repository at this point in the history

  28. some more linops type fixes

    @st--
    st-- committed Apr 13, 2023
    Configuration menu
    Copy the full SHA
    ff5211a View commit details
    Browse the repository at this point in the history

  29. Merge remote-tracking branch 'upstream/v0.6' into st/beartype

    @st--
    st-- committed Apr 13, 2023
    Configuration menu
    Copy the full SHA
    f3cc17e View commit details
    Browse the repository at this point in the history

Commits on Apr 17, 2023

  1. actually commit KeyArray and Scalar* types

    @st--
    st-- committed Apr 17, 2023
    Configuration menu
    Copy the full SHA
    af69a73 View commit details
    Browse the repository at this point in the history

  2. add beartype to pyproject

    @st--
    st-- committed Apr 17, 2023
    Configuration menu
    Copy the full SHA
    cc97fe2 View commit details
    Browse the repository at this point in the history

  3. from beartype.typing import ...

    @st--
    st-- committed Apr 17, 2023
    Configuration menu
    Copy the full SHA
    97d6fdd View commit details
    Browse the repository at this point in the history

  4. try to fix Self in gpjax/base/module

    @st--
    st-- committed Apr 17, 2023
    Configuration menu
    Copy the full SHA
    cd7c2ee View commit details
    Browse the repository at this point in the history

  5. fix _check_shape

    @st--
    st-- committed Apr 17, 2023
    Configuration menu
    Copy the full SHA
    8a62aae View commit details
    Browse the repository at this point in the history

  6. gpjax.objectives: always import from gps and variational_families

    @st--
    st-- committed Apr 17, 2023
    Configuration menu
    Copy the full SHA
    9359b05 View commit details
    Browse the repository at this point in the history

  7. Revert "gpjax.objectives: always import from gps and variational_fami… 

    …lies"

This reverts commit 9359b05.
    @st--
    st-- committed Apr 17, 2023
    Configuration menu
    Copy the full SHA
    c687cfe View commit details
    Browse the repository at this point in the history

  8. fix gpjax.objectives imported types

    @st--
    st-- committed Apr 17, 2023
    Configuration menu
    Copy the full SHA
    96d3d1e View commit details
    Browse the repository at this point in the history

  9. <...> | None not supported by beartype; replaced by Optional[<...>]

    @st--
    st-- committed Apr 17, 2023
    Configuration menu
    Copy the full SHA
    fc18652 View commit details
    Browse the repository at this point in the history

  10. gpjax.datasets: cannot specify strict array shape AND rely on _check_… 

    …shape
    @st--
    st-- committed Apr 17, 2023
    Configuration menu
    Copy the full SHA
    4cbe562 View commit details
    Browse the repository at this point in the history

  11. our tfd.Distribution subclassing requires the fix introduced in jaxty… 

    …ping 0.2.15
    @st--
    st-- committed Apr 17, 2023
    Configuration menu
    Copy the full SHA
    16006ac View commit details
    Browse the repository at this point in the history

  12. need to import base first!

    @st--
    st-- committed Apr 17, 2023
    Configuration menu
    Copy the full SHA
    c95badb View commit details
    Browse the repository at this point in the history

  13. bugfix

    @st--
    st-- committed Apr 17, 2023
    Configuration menu
    Copy the full SHA
    3c8969d View commit details
    Browse the repository at this point in the history

  14. AbstractKernel: string for forward references

    @st--
    st-- committed Apr 17, 2023
    Configuration menu
    Copy the full SHA
    243e22f View commit details
    Browse the repository at this point in the history

Commits on Apr 19, 2023

  1. remove from __future__ import annotations

    @st--
    st-- committed Apr 19, 2023
    Configuration menu
    Copy the full SHA
    0c3ae8a View commit details
    Browse the repository at this point in the history

  2. fix type annotations to make up for changes in 0c3ae8a

    @st--
    st-- committed Apr 19, 2023
    Configuration menu
    Copy the full SHA
    cd6cf66 View commit details
    Browse the repository at this point in the history

Commits on Apr 21, 2023

  1. pytree map functions may take a non-Module argument

    @st--
    st-- committed Apr 21, 2023
    Configuration menu
    Copy the full SHA
    c31885e View commit details
    Browse the repository at this point in the history

  2. ScalarFloat

    @st--
    st-- committed Apr 21, 2023
    Configuration menu
    Copy the full SHA
    7083d08 View commit details
    Browse the repository at this point in the history

  3. VecNOrMatNM

    @st--
    st-- committed Apr 21, 2023
    Configuration menu
    Copy the full SHA
    692d337 View commit details
    Browse the repository at this point in the history

  4. remove unnecessary / buggy methods

    @st--
    st-- committed Apr 21, 2023
    Configuration menu
    Copy the full SHA
    1b4ae52 View commit details
    Browse the repository at this point in the history

  5. more ScalarFloat

    @st--
    st-- committed Apr 21, 2023
    Configuration menu
    Copy the full SHA
    7286cd0 View commit details
    Browse the repository at this point in the history

  6. ScalarFloat

    @st--
    st-- committed Apr 21, 2023
    Configuration menu
    Copy the full SHA
    6e55323 View commit details
    Browse the repository at this point in the history

  7. type fixes

    @st--
    st-- committed Apr 21, 2023
    Configuration menu
    Copy the full SHA
    8159a2d View commit details
    Browse the repository at this point in the history

  8. fix shape type

    @st--
    st-- committed Apr 21, 2023
    Configuration menu
    Copy the full SHA
    62c7a69 View commit details
    Browse the repository at this point in the history

Commits on Apr 24, 2023

  1. fix one KeyArray

    @st--
    st-- committed Apr 24, 2023
    Configuration menu
    Copy the full SHA
    3a8814a View commit details
    Browse the repository at this point in the history

  2. more ScalarFloat corrections in kernels

    @st--
    st-- committed Apr 24, 2023
    Configuration menu
    Copy the full SHA
    25ed38f View commit details
    Browse the repository at this point in the history

  3. fix test_stationary accordingly for ScalarFloat params

    @st--
    st-- committed Apr 24, 2023
    Configuration menu
    Copy the full SHA
    76dd035 View commit details
    Browse the repository at this point in the history

  4. fix return type

    @st--
    st-- committed Apr 24, 2023
    Configuration menu
    Copy the full SHA
    2ba27b2 View commit details
    Browse the repository at this point in the history

  5. ScalarInt for Polynomial kernel and fix test for Scalar* params

    @st--
    st-- committed Apr 24, 2023
    Configuration menu
    Copy the full SHA
    33a3e7b View commit details
    Browse the repository at this point in the history

  6. fix mock in test_abstract_variational_family

    @st--
    st-- committed Apr 24, 2023
    Configuration menu
    Copy the full SHA
    4dd9ed5 View commit details
    Browse the repository at this point in the history

  7. fix link_function and variational_expectations shape annotations

    @st--
    st-- committed Apr 24, 2023
    Configuration menu
    Copy the full SHA
    54ef19c View commit details
    Browse the repository at this point in the history

  8. minor test fix

    @st--
    st-- committed Apr 24, 2023
    Configuration menu
    Copy the full SHA
    3a21676 View commit details
    Browse the repository at this point in the history

  9. fix exception test for beartype

    @st--
    st-- committed Apr 24, 2023
    Configuration menu
    Copy the full SHA
    551f055 View commit details
    Browse the repository at this point in the history

  10. fix Constant mean function

    @st--
    st-- committed Apr 24, 2023
    Configuration menu
    Copy the full SHA
    6fc3a55 View commit details
    Browse the repository at this point in the history

  11. base_kernel as kwarg in test_approximations

    @st--
    st-- committed Apr 24, 2023
    Configuration menu
    Copy the full SHA
    bf2a483 View commit details
    Browse the repository at this point in the history

  12. rename func to test_ so it actually gets collected

    @st--
    st-- committed Apr 24, 2023
    Configuration menu
    Copy the full SHA
    ac70773 View commit details
    Browse the repository at this point in the history

  13. mark test_graph_kernel as broken

    @st--
    st-- committed Apr 24, 2023
    Configuration menu
    Copy the full SHA
    e2b6a88 View commit details
    Browse the repository at this point in the history

  14. fix LinearOperator DTypeT

    @st--
    st-- committed Apr 24, 2023
    Configuration menu
    Copy the full SHA
    985d554 View commit details
    Browse the repository at this point in the history

  15. Revert "fix Constant mean function" 

    This reverts commit 6fc3a55.
    @st--
    st-- committed Apr 24, 2023
    Configuration menu
    Copy the full SHA
    7ea37d2 View commit details
    Browse the repository at this point in the history

  16. fix test_mean_functions instead

    @st--
    st-- committed Apr 24, 2023
    Configuration menu
    Copy the full SHA
    760d470 View commit details
    Browse the repository at this point in the history

  17. fix one more bug in RFF test

    @st--
    st-- committed Apr 24, 2023
    Configuration menu
    Copy the full SHA
    fcf6572 View commit details
    Browse the repository at this point in the history

  18. Self

    @st--
    st-- committed Apr 24, 2023
    Configuration menu
    Copy the full SHA
    78397d5 View commit details
    Browse the repository at this point in the history

  19. relax fit objective type

    @st--
    st-- committed Apr 24, 2023
    Configuration menu
    Copy the full SHA
    67f4c94 View commit details
    Browse the repository at this point in the history

Commits on Apr 25, 2023

  1. rename gpjax.utils -> gpjax.typing

    @st--
    st-- committed Apr 25, 2023
    Configuration menu
    Copy the full SHA
    68d6371 View commit details
    Browse the repository at this point in the history

  2. Kernel = Any -> string forward reference

    @st--
    st-- committed Apr 25, 2023
    Configuration menu
    Copy the full SHA
    65a9bfb View commit details
    Browse the repository at this point in the history

  3. relax Gaussian.predict type annotation to include GaussianDistribution

    @st--
    st-- committed Apr 25, 2023
    Configuration menu
    Copy the full SHA
    c439a93 View commit details
    Browse the repository at this point in the history

  4. our own Array type that accepts both JAX and Numpy arrays

    @st--
    st-- committed Apr 25, 2023
    Configuration menu
    Copy the full SHA
    edaf2cb View commit details
    Browse the repository at this point in the history

  5. some Float -> Num relaxations for graph kernel...

    @st--
    st-- committed Apr 25, 2023
    Configuration menu
    Copy the full SHA
    bb4d51f View commit details
    Browse the repository at this point in the history

  6. ScalarFloat for GraphKernel hyperparams

    @st--
    st-- committed Apr 25, 2023
    Configuration menu
    Copy the full SHA
    ea1537d View commit details
    Browse the repository at this point in the history

  7. fix type hints to what happens (even if it seems wrong)

    @st--
    st-- committed Apr 25, 2023
    Configuration menu
    Copy the full SHA
    495c649 View commit details
    Browse the repository at this point in the history

  8. type relaxation for deep_kernels.pct.py

    @st--
    st-- committed Apr 25, 2023
    Configuration menu
    Copy the full SHA
    dca6799 View commit details
    Browse the repository at this point in the history

  9. some more minor consistency fixes

    @st--
    st-- committed Apr 25, 2023
    Configuration menu
    Copy the full SHA
    f26d52e View commit details
    Browse the repository at this point in the history

  10. bugfix

    @st--
    st-- committed Apr 25, 2023
    Configuration menu
    Copy the full SHA
    23efd03 View commit details
    Browse the repository at this point in the history

  11. Update examples/graph_kernels.pct.py 

    Co-authored-by: st-- <st--@users.noreply.github.com>
Signed-off-by: Thomas Pinder <tompinder@live.co.uk>
    @thomaspinder @st--
    thomaspinder and st-- authored Apr 25, 2023
    Configuration menu
    Copy the full SHA
    1077efa View commit details
    Browse the repository at this point in the history

  12. Update gpjax/dataset.py 

    Allow for integer responses

Co-authored-by: st-- <st--@users.noreply.github.com>
Signed-off-by: Thomas Pinder <tompinder@live.co.uk>
    @thomaspinder @st--
    thomaspinder and st-- authored Apr 25, 2023
    Configuration menu
    Copy the full SHA
    c292c7e View commit details
    Browse the repository at this point in the history

Commits on Apr 26, 2023

  1. jaxtyping import hook for notebooks

    @st--
    st-- committed Apr 26, 2023
    Configuration menu
    Copy the full SHA
    7e16487 View commit details
    Browse the repository at this point in the history

  2. conftest.py to apply jaxtyping import hook before loading tests

    @st--
    st-- committed Apr 26, 2023
    Configuration menu
    Copy the full SHA
    6620802 View commit details
    Browse the repository at this point in the history

  3. remove import hook from gpjax/__init__

    @st--
    st-- committed Apr 26, 2023
    Configuration menu
    Copy the full SHA
    bc70fdf View commit details
    Browse the repository at this point in the history

  4. Merge branch 'v0.6' of https://github.com/JaxGaussianProcesses/GPJax … 

    …into st/beartype
    @st--
    st-- committed Apr 26, 2023
    Configuration menu
    Copy the full SHA
    249c9d0 View commit details
    Browse the repository at this point in the history

  5. Update gpjax/dataset.py 

    Signed-off-by: st-- <st--@users.noreply.github.com>
    @st--
    st-- authored Apr 26, 2023
    Configuration menu
    Copy the full SHA
    c4a8af8 View commit details
    Browse the repository at this point in the history

  6. Update gpjax/dataset.py 

    Co-authored-by: st-- <st--@users.noreply.github.com>
Signed-off-by: Thomas Pinder <tompinder@live.co.uk>
    @thomaspinder @st--
    thomaspinder and st-- authored Apr 26, 2023
    Configuration menu
    Copy the full SHA
    62abd1a View commit details
    Browse the repository at this point in the history

  7. fix tests of shape checks now that we have beartype

    @st--
    st-- committed Apr 26, 2023
    Configuration menu
    Copy the full SHA
    42e21fb View commit details
    Browse the repository at this point in the history