That's a tricky one since learn implementation are currently not cleanly separated (violate Liskov substitution) and rely on kwargs in order to remain functional. E.g. some implementations of learn take the repeat kwarg (the on-policy ones), and some don't.

Generally, the control flow between trainer.update_policy and policy.learn may need to be redesigned or at least improved