__init__.py

from stable_baselines.a2c import A2C
from stable_baselines.acer import ACER
from stable_baselines.acktr import ACKTR
from stable_baselines.deepq import DQN
from stable_baselines.deepq_kpi import DQN
from stable_baselines.deepq_hpi import DQN
from stable_baselines.her import HER
from stable_baselines.ppo2 import PPO2
from stable_baselines.td3 import TD3
from stable_baselines.sac import SAC


# Load mpi4py-dependent algorithms only if mpi is installed.
try:
    import mpi4py
except ImportError:
    mpi4py = None

if mpi4py is not None:
    from stable_baselines.ddpg import DDPG
    from stable_baselines.gail import GAIL
    from stable_baselines.ppo1 import PPO1
    from stable_baselines.trpo_kpi import TRPO
del mpi4py

__version__ = "2.7.0"