For your last question: no. You can pass the id of your current gist to update it.
This is all looking pretty good and almost ready to be joined to the library. For the tests requiring training, I might pick just one and couple it with our integration test because we don’t want too many training tests (those take a long time on CPU). The other would be marked as slow and only run locally before doing a release or an important PR.
Thanks a lot for all your work on this!