__init__.py

# general
from .q_learning import DQN, RainbowDQN, QRDQN, IQN, FQF, DRQN, C51DQN, BDQ
from .qac import QAC, DiscreteQAC
from .pdqn import PDQN
from .vac import VAC
from .bc import DiscreteBC, ContinuousBC
from .pg import PG
# algorithm-specific
from .ppg import PPG
from .qmix import Mixer, QMix
from .collaq import CollaQ
from .wqmix import WQMix
from .coma import COMA
from .atoc import ATOC
from .sqn import SQN
from .acer import ACER
from .qtran import QTran
from .mavac import MAVAC
from .ngu import NGU
from .qac_dist import QACDIST
from .maqac import MAQAC, ContinuousMAQAC
from .madqn import MADQN
from .vae import VanillaVAE
from .decision_transformer import DecisionTransformer
from .procedure_cloning import ProcedureCloningMCTS