from sb3_trl.trl_pg.policies import CnnPolicy, MlpPolicy, MultiInputPolicy
from sb3_trl.trl_pg.trl_pg import TRL_PG