From d5e53a87db0a0ac84388d97c1f4e6eafd610a656 Mon Sep 17 00:00:00 2001 From: Dominik Roth Date: Tue, 16 Jan 2024 15:34:12 +0100 Subject: [PATCH] Expose TRPL class --- metastable_baselines2/__init__.py | 2 ++ 1 file changed, 2 insertions(+) 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", ]