Hylke

Thanks for your detailed review. I'll get back to you soon. Hylke On Sun, 13 Dec 2020, 00:23 Jacob Schreiber, wrote: > *@jmschrei* commented on this pull request. > >...

Hi Jacob, I have made all the requested changes, plus I updated the `BayesClassifier` model, which was missing from the previous commit. I hope this looks okay for you. Kind...

Can the failing CI be related to the new release of joblib (it looks like https://github.com/scikit-learn-contrib/hdbscan/issues/436)?

Thanks for your prompt response. I've created a pull request which should address both points: https://github.com/AlexandrovLab/SigProfilerMatrixGenerator/pull/81 Thanks! Hylke

Any thoughts on how to checkpoint a Flax `TrainState` with the new `CheckpointManager` API? I gave [it a go on StackOverflow](https://stackoverflow.com/questions/78033458/using-orbax-to-checkpoint-flax-trainstate-with-new-checkpointmanager-api), but without success.

Yes, same problem on `flax==0.7.5`. Strangely enough, I haven't been able to replicate my problem on Ubuntu 23.10 Docker. That is, the following seems to work fine, although it should...

Nope, same problem as HEAD. Anything that I can do to further clarify?