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