diff --git a/environment/breakout_env.py b/environment/breakout_env.py
index 9d9293e..a5cf731 100644
--- a/environment/breakout_env.py
+++ b/environment/breakout_env.py
@@ -4,15 +4,14 @@
 
 import gymnasium as gym
 import numpy as np
-import torch
-import ale_py  # Import to trigger auto-registration
+import ale_py  # noqa: F401  # Import to trigger auto-registration
 from gymnasium.wrappers import (
     AtariPreprocessing, 
     RecordEpisodeStatistics,
     FrameStackObservation
 )
-from gymnasium.vector import AsyncVectorEnv, SyncVectorEnv
-from typing import Tuple, Any
+from gymnasium.vector import SyncVectorEnv
+from typing import Tuple