From 1f0748e49f05a8cce82b8678c1fb17c76aa83514 Mon Sep 17 00:00:00 2001
From: Dominik Roth <dominik.roth.dev@gmail.com>
Date: Tue, 22 Aug 2023 00:15:47 +0200
Subject: [PATCH] Fix: Referenced original SAC instead of better one

---
 sbBrix/__init__.py     | 5 -----
 sbBrix/sac/__init__.py | 2 +-
 2 files changed, 1 insertion(+), 6 deletions(-)

diff --git a/sbBrix/__init__.py b/sbBrix/__init__.py
index b3b8f88..d52b3da 100644
--- 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",
 ]
-
diff --git a/sbBrix/sac/__init__.py b/sbBrix/sac/__init__.py
index 813d1fb..59a45df 100644
--- 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