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.

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

Update notebooks. #447

Merged
merged 4 commits into from
Mar 22, 2024
Merged

Update notebooks. #447

merged 4 commits into from
Mar 22, 2024

Conversation

daniel-dodd
Copy link
Member

Type of changes

  • Bug fix
  • New feature
  • Documentation / docstrings
  • Tests
  • Other

Checklist

  • I've formatted the new code by running poetry run pre-commit run --all-files --show-diff-on-failure before committing.
  • I've added tests for new code.
  • I've added docstrings for the new code.

Description

Please describe your changes here. If this fixes a bug, please link to the issue, if possible.

Issue Number: N/A

@daniel-dodd daniel-dodd requested a review from frazane March 22, 2024 12:30
@frazane frazane merged commit 30ed414 into flax-backend-normalclasses Mar 22, 2024
2 of 11 checks passed
thomaspinder pushed a commit that referenced this pull request Jul 9, 2024
* Update yacht.py

* Update likelihoods_guide.py

* Revert "Update likelihoods_guide.py"

This reverts commit 5f51cfe.

* Update oceanmodelling.py
thomaspinder added a commit that referenced this pull request Aug 16, 2024
* add flax v0.8.0 to deps, temporarily from github main branch

* main gps objects as nnx modules

* integrators as nnx dataclasses and some static typing refactoring

* likelihoods as nnx dataclasses modules and some static typing refactoring

* small refactoring

* mean functions as nnx dataclasses modules and some refactoring

* bugfix

* objectives as nnx dataclasses modules

* variational families with nnx

* kernels base with nnx

* wip stationary kernels

* wip nonstationary kernels

* wip non euclidean kernels

* computations with nnx

* rff with nnx

* bugfix

* stationary kernels as normal classes

* nonstationary kernels as normal classes

* noneuclidean kernels as normal classes

* rff as standard class + stationary kernel abstract class for static typing

* started work on parameters

* more objects as normal classes

* gps as normal classes

* integrators as normal classes

* dataset is not a pytree

* removed superfluous inits

* register dataset as pytree

* use parameters here and there

* set active_dims default to 1

* start working on tests

* active_dims defaults to None

* rewrite objectives as functions

Co-authored-by: Daniel Dodd <[email protected]>

* black + isort

* remove objective from cite

* fix dataset repr

* pass tests for variational families

* active_dims defaults to None

* use generic Objective type

* small fixes

* make 'active_dims' required parameter, fix static typing and beartype for parameters, rewrite and pass tests for stationary kernels

* pass tests/test_kernels/test_computation.py

* rewrite tests for nonstationary kernels + pass tests

* adapt to nnx's explicit variables + miscellaneous fixes

* rewrite of objectives as simple functions, [WIP] started rewriting tests

* rewrite and pass tests for objectives

* rewrite fit function

* remove gpjax.base module

* remove base module tests

* rewrite and pass tests for fit

* finish kernels and pass all tests

* pass all tests except decision making

* pass all tests 🚀

* update and run classification notebook (python cells)

* pass doctests

* pass integration tests, more checks to parameters

* linting and formatting

* update barycentres and classification examples

* update project files

* update ruff and make it happy

* lint + format all doc examples

* [skip ci] change how dimensions are specified for kernels, update kernel tests

* [skip ci] api reference looks pretty now, implemented template pattern, improved docstrings

* [skip ci] wip - fixing math rendering in documentation - almost there

* Update notebooks. (#447)

* Update yacht.py

* Update likelihoods_guide.py

* Revert "Update likelihoods_guide.py"

This reverts commit 5f51cfe.

* Update oceanmodelling.py

* Update likelihoods.py (#446)

* Update likelihoods.py

* Update likelihoods.py

* Update likelihoods.py

* Adding tagged parameters and updated notebooks

* Update likelihoods.py (#446)

* Update likelihoods.py

* Update likelihoods.py

* Update likelihoods.py

* Update notebooks

* Fix linting

* Fix missing dep.

* Fix integration test

* Readd docs deps

* Fix docstrings

* Update lockfile

* Update parameter refs

* Fix broken tests

* Remove PyTrees doc

* Failing split order

* NNX update

* add flax v0.8.0 to deps, temporarily from github main branch

* main gps objects as nnx modules

* integrators as nnx dataclasses and some static typing refactoring

* likelihoods as nnx dataclasses modules and some static typing refactoring

* small refactoring

* mean functions as nnx dataclasses modules and some refactoring

* bugfix

* objectives as nnx dataclasses modules

* variational families with nnx

* kernels base with nnx

* wip stationary kernels

* wip nonstationary kernels

* wip non euclidean kernels

* computations with nnx

* rff with nnx

* bugfix

* stationary kernels as normal classes

* nonstationary kernels as normal classes

* noneuclidean kernels as normal classes

* rff as standard class + stationary kernel abstract class for static typing

* started work on parameters

* more objects as normal classes

* gps as normal classes

* integrators as normal classes

* dataset is not a pytree

* removed superfluous inits

* register dataset as pytree

* use parameters here and there

* set active_dims default to 1

* start working on tests

* active_dims defaults to None

* rewrite objectives as functions

Co-authored-by: Daniel Dodd <[email protected]>

* black + isort

* remove objective from cite

* fix dataset repr

* pass tests for variational families

* active_dims defaults to None

* use generic Objective type

* small fixes

* make 'active_dims' required parameter, fix static typing and beartype for parameters, rewrite and pass tests for stationary kernels

* pass tests/test_kernels/test_computation.py

* rewrite tests for nonstationary kernels + pass tests

* adapt to nnx's explicit variables + miscellaneous fixes

* rewrite of objectives as simple functions, [WIP] started rewriting tests

* rewrite and pass tests for objectives

* rewrite fit function

* remove gpjax.base module

* remove base module tests

* rewrite and pass tests for fit

* finish kernels and pass all tests

* pass all tests except decision making

* pass all tests 🚀

* update and run classification notebook (python cells)

* pass doctests

* pass integration tests, more checks to parameters

* linting and formatting

* update barycentres and classification examples

* update project files

* update ruff and make it happy

* lint + format all doc examples

* [skip ci] change how dimensions are specified for kernels, update kernel tests

* [skip ci] api reference looks pretty now, implemented template pattern, improved docstrings

* [skip ci] wip - fixing math rendering in documentation - almost there

* Update notebooks. (#447)

* Update yacht.py

* Update likelihoods_guide.py

* Revert "Update likelihoods_guide.py"

This reverts commit 5f51cfe.

* Update oceanmodelling.py

* Update likelihoods.py (#446)

* Update likelihoods.py

* Update likelihoods.py

* Update likelihoods.py

* Update notebooks

* Adding tagged parameters and updated notebooks

* Fix linting

* Fix missing dep.

* Fix integration test

* Readd docs deps

* Fix docstrings

* Update lockfile

* Update parameter refs

* Fix broken tests

* Remove PyTrees doc

* Failing split order

* NNX update

* rename static dir

* move examples dir in top level

* add _examples generated dir to gitignore

* update pyproject deps

* update mkdocs config

* add examples generation script

* adapt relative paths in md files

* Update Ruff and incorporate changes

* update github workflow for building doc, without executing notebookf for now

* Add backend doc

* Add backend doc

* Add backend doc

* Add replace to transform

* Merge with main

* Update parameters docstring

* Respond to comments

* Fix e2e tests

* Fix mplstyle refs

* bump deps

* Update poetry

* Update poetry

* Fix shutil

* Drop flax base

* add scikit-learn dependency for docs

* bugfix: change directory before running jupytext

* use local mpl style file

* do not use MCMC for classification (it is *very* slow)

* [skip-ci] update github workflows for docs

* Fix split

* Fix split

* Fix split

* Fix xdoctest

* Fix doc

* Add serial build

* Update parameters transform and backend doc

* Update parameters transform and backend doc

* Bump Python

---------

Signed-off-by: Thomas Pinder <[email protected]>
Co-authored-by: Daniel Dodd <[email protected]>
Co-authored-by: Daniel Dodd <[email protected]>
Co-authored-by: Thomas Pinder <[email protected]>
Co-authored-by: Thomas-Christie <[email protected]>
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants