From 3d2b7dfc8f113a16e958fd295a4a2c6eb525a7de Mon Sep 17 00:00:00 2001 From: Dominik Roth Date: Tue, 16 Jan 2024 15:35:18 +0100 Subject: [PATCH] import metastable_projections corretly when accesing via string name --- metastable_baselines2/trpl/trpl.py | 1 + 1 file changed, 1 insertion(+) diff --git a/metastable_baselines2/trpl/trpl.py b/metastable_baselines2/trpl/trpl.py index 7f91a87..ea16a50 100644 --- a/metastable_baselines2/trpl/trpl.py +++ b/metastable_baselines2/trpl/trpl.py @@ -13,6 +13,7 @@ from stable_baselines3.common.type_aliases import GymEnv, MaybeCallback, Schedul from stable_baselines3.common.utils import explained_variance, get_schedule_fn from metastable_projections import BaseProjectionLayer, FrobeniusProjectionLayer, WassersteinProjectionLayer, KLProjectionLayer +import metastable_projections SelfTRPL = TypeVar("SelfTRPL", bound="TRPL")