Fix importing

msosav · Jan 11, 2025 · ce8fe2a · ce8fe2a
1 parent b089bc7
commit ce8fe2a
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 7 deletions.
diff --git a/config/gym.py b/config/gym.py
@@ -1,6 +1,6 @@
 import gymnasium as gym
 import numpy as np
-from gymnasium.spaces import Box, Discrete, Dict
+from gymnasium.spaces import Box, Dict, Discrete
 from pyboy import PyBoy
 from pyboy.utils import WindowEvent
 

diff --git a/main.py b/main.py
@@ -2,9 +2,6 @@
 
 from stable_baselines3 import PPO
 
-from stable_baselines3.common.vec_env import SubprocVecEnv
-from stable_baselines3.common import env_checker
-
 from utils import CheckpointAndLoggingCallback, PreprocessEnv
 
 if __name__ == "__main__":

diff --git a/utils.py b/utils.py
@@ -1,9 +1,10 @@
 import os
 
 import gymnasium as gym
-
-from gymnasium.wrappers import gray_scale_observation
-from stable_baselines3.common.callbacks import BaseCallback  # For saving models
+import numpy as np
+from gymnasium.spaces import Box
+from stable_baselines3.common.callbacks import \
+    BaseCallback
 from stable_baselines3.common.vec_env import DummyVecEnv, VecFrameStack
 from gymnasium.spaces import Box
 import numpy as np