From 87709338a02a0ef909abcc569460be4b2dbac7e3 Mon Sep 17 00:00:00 2001 From: cromac Date: Mon, 5 Aug 2024 17:55:01 +0200 Subject: [PATCH] Fixing imports --- experiments/agents/bot/bot.py | 2 +- experiments/agents/drrn/drrn.py | 2 +- experiments/agents/ppo/base_ppo_agent.py | 2 +- experiments/agents/random_agent/random_agent.py | 2 +- experiments/train_symbolic_ppo.py | 2 +- 5 files changed, 5 insertions(+), 5 deletions(-) diff --git a/experiments/agents/bot/bot.py b/experiments/agents/bot/bot.py index 1637040..50e4ddc 100644 --- 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 index 108da62..3405111 100644 --- 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 index 2546330..9f72fec 100644 --- 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 index c410e76..6655ec6 100644 --- 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 index e7edf70..fa8d505 100755 --- 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__":