diff --git a/sbBrix/sac/sac.py b/sbBrix/sac/sac.py index 52213c2..b10bf55 100644 --- a/sbBrix/sac/sac.py +++ b/sbBrix/sac/sac.py @@ -8,7 +8,7 @@ from torch.nn import functional as F from stable_baselines3.common.buffers import ReplayBuffer from stable_baselines3.common.noise import ActionNoise # from stable_baselines3.common.off_policy_algorithm import OffPolicyAlgorithm -from common.off_policy_algorithm import BetterOffPolicyAlgorithm +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