Change agents to work with sb3 version > 1.6

Agents need to be adjusted to only supply policy_base parameter, when it is required by the stable_baselines version.

Close #128 (closed)