From 226e53d0cdb44a303cd9181386d4f7d687160288 Mon Sep 17 00:00:00 2001 From: ll7 Date: Wed, 6 Mar 2024 14:22:29 +0100 Subject: [PATCH] Refactor imports in sb3_test.py --- tests/sb3_test.py | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/tests/sb3_test.py b/tests/sb3_test.py index 3ac1f51..b390f0d 100644 --- 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():