From b67bd857b80de2d638f239427140ef1476f02c77 Mon Sep 17 00:00:00 2001 From: RickFqt Date: Tue, 12 Nov 2024 07:31:50 -0300 Subject: [PATCH] fix: Fix imports --- experiments/collectables_check_env.py | 2 +- experiments/solves/solve_collectables_sb3.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/experiments/collectables_check_env.py b/experiments/collectables_check_env.py index 3071885..3e36fb6 100644 --- 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 index 5e3256a..26e2206 100644 --- 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