fix: Fix imports

UFRN-URNAI · Nov 12, 2024 · b67bd85 · b67bd85
1 parent 309a735
commit b67bd85
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/experiments/collectables_check_env.py b/experiments/collectables_check_env.py
@@ -8,9 +8,9 @@
 from pysc2.env import sc2_env
 from stable_baselines3.common.env_checker import check_env
 
+from urnai.environments.stablebaselines3.custom_env import CustomEnv
 from urnai.sc2.actions.collectables import CollectablesActionSpace
 from urnai.sc2.environments.sc2environment import SC2Env
-from urnai.sc2.environments.stablebaselines3.custom_env import CustomEnv
 from urnai.sc2.rewards.collectables import CollectablesReward
 from urnai.sc2.states.collectables import CollectablesMethod, CollectablesState
 

diff --git a/experiments/solves/solve_collectables_sb3.py b/experiments/solves/solve_collectables_sb3.py
@@ -7,9 +7,9 @@
 from pysc2.env import sc2_env
 from stable_baselines3 import PPO
 
+from urnai.environments.stablebaselines3.custom_env import CustomEnv
 from urnai.sc2.actions.collectables import CollectablesActionSpace
 from urnai.sc2.environments.sc2environment import SC2Env
-from urnai.sc2.environments.stablebaselines3.custom_env import CustomEnv
 from urnai.sc2.rewards.collectables import CollectablesReward
 from urnai.sc2.states.collectables import CollectablesMethod, CollectablesState