diff --git a/metastable_baselines2/__init__.py b/metastable_baselines2/__init__.py index 07d11b3..394ebeb 100644 --- a/metastable_baselines2/__init__.py +++ b/metastable_baselines2/__init__.py @@ -1,7 +1,9 @@ from metastable_baselines2.ppo import PPO from metastable_baselines2.sac import SAC +from metastable_baselines2.trpl import TRPL __all__ = [ "PPO", "SAC", + "TRPL", ]