Update StableBaselines3 and switch from Gym to Gymnasium

These changes have to be done simultaneously because StableBaselines3 2.0 uses Gymnasium instead of Gym.