From 28a518fe9dd74e8972aa96b78c9832a3fce5810d Mon Sep 17 00:00:00 2001 From: Dominik Roth Date: Tue, 22 Aug 2023 00:19:13 +0200 Subject: [PATCH] Fix: Referenced old SACPolicy --- sbBrix/sac/sac.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/sbBrix/sac/sac.py b/sbBrix/sac/sac.py index b10bf55..0979c1d 100644 --- a/sbBrix/sac/sac.py +++ b/sbBrix/sac/sac.py @@ -12,7 +12,7 @@ from ..common.off_policy_algorithm import BetterOffPolicyAlgorithm from stable_baselines3.common.policies import BasePolicy from stable_baselines3.common.type_aliases import GymEnv, MaybeCallback, Schedule from stable_baselines3.common.utils import get_parameters_by_name, polyak_update -from stable_baselines3.sac.policies import CnnPolicy, MlpPolicy, MultiInputPolicy, SACPolicy +from ..common.policies import MlpPolicy, SACPolicy SelfSAC = TypeVar("SelfSAC", bound="SAC")