Bump pytorch-lightning to v2.0.0, make NOTMAD compatible, rerun doc notebooks

cnellington / Contextualized

An SKLearn-style toolbox for estimating and analyzing models, distributions, and functions with context-specific parameters.

http://contextualized.ml/

GNU General Public License v3.0

64 stars 7 forks source link

Closed blengerich closed 1 year ago

blengerich commented 1 year ago

Bumps pytorch-lightning to v2.0.0, which
Requires changing how NOTMAD deals with training_epoch_end function (renamed on_train_epoch_end)
Reruns all doc notebooks (fixes #194 )

review-notebook-app[bot] commented 1 year ago

Check out this pull request on

See visual diffs & provide feedback on Jupyter Notebooks.

Powered by ReviewNB

cnellington commented 1 year ago

Code looks good but tests failed, looks related to the lightning version upgrade: I see "GraphTrainer has no attribute tune" in one test.