From cb48badcff0e601b0f7d83a6844277b7016d828a Mon Sep 17 00:00:00 2001 From: Dominik Roth Date: Wed, 28 Aug 2024 11:32:07 +0200 Subject: [PATCH] Adapt init --- fancy_rl/__init__.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/fancy_rl/__init__.py b/fancy_rl/__init__.py index 323df6b..011f43e 100644 --- a/fancy_rl/__init__.py +++ b/fancy_rl/__init__.py @@ -4,6 +4,7 @@ try: except ImportError: pass -from fancy_rl.algos import PPO +from fancy_rl.algos import PPO, TRPL, VLEARN +from fancy_rl.projections import get_projection -__all__ = ["PPO"] \ No newline at end of file +__all__ = ["PPO", "TRPL", "VLEARN", "get_projection"] \ No newline at end of file