stable_baselines3 versions > 1.6 not working

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