Refactor imports in sb3_test.py

ll7 · Mar 6, 2024 · 226e53d · 226e53d
1 parent 43aa423
commit 226e53d
Showing 1 changed file with 1 addition and 4 deletions.
diff --git a/tests/sb3_test.py b/tests/sb3_test.py
@@ -1,14 +1,11 @@
 import os
 
 from stable_baselines3.ppo.ppo import PPO
-from robot_sf.robot_env import RobotEnv
-from robot_sf.feature_extractor import DynamicsExtractor
-
-from stable_baselines3 import PPO
 from stable_baselines3.common.env_util import make_vec_env
 from stable_baselines3.common.vec_env import SubprocVecEnv
 
 from robot_sf.robot_env import RobotEnv
+from robot_sf.feature_extractor import DynamicsExtractor
 
 
 def test_can_load_model_snapshot():