dodox/metastable-baselines2
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

import metastable_projections corretly when accesing via string name

Browse Source
This commit is contained in:
Dominik Moritz Roth 2024-01-16 15:35:18 +01:00
parent d5e53a87db
commit 3d2b7dfc8f
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
metastable_baselines2/trpl/trpl.py
View File

@ -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")