The Trainer currently uses a bunch of functions injected into it and also tracks a bunch of specific variables which is more restrictive than necessary. See e.g. discussion in #1202 (comment)

The situation can likely be improved by switching to something more similar to the Strategy pattern, but this would constitute a rather large breaking change. Might be addressed by release 2.0 or before that