Fix: Referenced original SAC instead of better one

2023-08-22 00:15:47 +02:00 · 2023-08-22 00:15:47 +02:00 · 1f0748e49f
commit 1f0748e49f
parent 8384d32287
2 changed files with 1 additions and 6 deletions
--- a/sbBrix/init.py
+++ b/sbBrix/init.py
@ -1,7 +1,3 @@
-import os
-
-import numpy as np
-
 from sbBrix.ppo import PPO
 from sbBrix.sac import SAC

@ -9,4 +5,3 @@ __all__ = [
    "PPO",
    "SAC",
 ]
-
--- a/sbBrix/sac/init.py
+++ b/sbBrix/sac/init.py
@ -1 +1 @@
-from stable_baselines3.sac.sac import SAC
+from sbBrix.sac.sac import SAC