From 74a6fa85e10fe3efb59d25de7aa9f2a6a380d330 Mon Sep 17 00:00:00 2001 From: Jeroen Oudshoorn Date: Sat, 7 Oct 2023 00:42:01 +0200 Subject: [PATCH] v2.4.7 Signed-off-by: Jeroen Oudshoorn --- pwnagotchi/ai/__init__.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/pwnagotchi/ai/__init__.py b/pwnagotchi/ai/__init__.py index b5821819..073f699e 100644 --- a/pwnagotchi/ai/__init__.py +++ b/pwnagotchi/ai/__init__.py @@ -21,12 +21,12 @@ def load(config, agent, epoch, from_disk=True): SB_BACKEND = "stable_baselines3" try: - from baselines.common.vec_env import DummyVecEnv - from baselines.a2c import a2c + from stable_baselines3.common.vec_env import DummyVecEnv + from stable_baselines3.a2c import a2c logging.debug("[ai] A2C imported in %.2fs" % (time.time() - start)) start = time.time() - from baselines.a2c.policies import MlpPolicy + from stable_baselines3.a2c.policies import MlpPolicy logging.debug("[ai] MlpPolicy imported in %.2fs" % (time.time() - start)) SB_A2C_POLICY = MlpPolicy