Some updates. We finally have some learner level unittests. Separated inference from model export, so we can test this feature in isolation.