Fixing imports

flowersteam · Aug 5, 2024 · 8770933 · 8770933
1 parent a2f9ce5
commit 8770933
Show file tree

Hide file tree

Showing 5 changed files with 5 additions and 5 deletions.
diff --git a/experiments/agents/bot/bot.py b/experiments/agents/bot/bot.py
@@ -4,7 +4,7 @@
 from tqdm import tqdm
 import numpy as np
 
-from experiments.agents.base_agent import BaseAgent
+from agents.base_agent import BaseAgent
 
 class BotAgent(BaseAgent):
     def __init__(self, envs, subgoals):

diff --git a/experiments/agents/drrn/drrn.py b/experiments/agents/drrn/drrn.py
@@ -19,7 +19,7 @@
 
 import babyai.rl
 
-from experiments.agents.base_agent import BaseAgent
+from agents.base_agent import BaseAgent
 
 # Accelerate
 from accelerate import Accelerator

diff --git a/experiments/agents/ppo/base_ppo_agent.py b/experiments/agents/ppo/base_ppo_agent.py
@@ -1,4 +1,4 @@
-from experiments.agents.base_agent import BaseAgent
+from agents.base_agent import BaseAgent
 
 from babyai.rl.utils.supervised_losses import ExtraInfoCollector
 

diff --git a/experiments/agents/random_agent/random_agent.py b/experiments/agents/random_agent/random_agent.py
@@ -1,7 +1,7 @@
 import numpy as np
 from tqdm import tqdm
 
-from experiments.agents.base_agent import BaseAgent
+from agents.base_agent import BaseAgent
 
 class Random_agent(BaseAgent):
     def __init__(self, envs, subgoals):

diff --git a/experiments/train_symbolic_ppo.py b/experiments/train_symbolic_ppo.py
@@ -25,7 +25,7 @@
 # from subtask_prediction import SubtaskPrediction, SubtaskDataset
 from colorama import Fore, Back, Style
 
-from experiments.agents.ppo.symbolic_ppo_agent import SymbolicPPOAgent
+from agents.ppo.symbolic_ppo_agent import SymbolicPPOAgent
 
 if __name__ == "__main__":