From 0162a368242c5b0ca7442212b1c5658600c0cc1c Mon Sep 17 00:00:00 2001 From: Dominik Roth Date: Tue, 19 Jul 2022 10:08:14 +0200 Subject: [PATCH] Renamed TRL_SAC => SAC --- metastable_baselines/sac/__init__.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/metastable_baselines/sac/__init__.py b/metastable_baselines/sac/__init__.py index c0e01b7..5250559 100644 --- a/metastable_baselines/sac/__init__.py +++ b/metastable_baselines/sac/__init__.py @@ -1,2 +1,2 @@ -from sb3_trl.trl_sac.policies import CnnPolicy, MlpPolicy, MultiInputPolicy -from sb3_trl.trl_sac.trl_sac import TRL_SAC +from ..sac.policies import CnnPolicy, MlpPolicy, MultiInputPolicy +from ..sac.sac import SAC